Pill Classification Image Dataset

Background Information

This dataset was curated and annotated by Mohamed Attia.

The original dataset (v1) is composed of 451 images of various pills that are present on a large variety of surfaces and objects.
Example of an Image from the Dataset

The dataset is available under the Public License.

Getting Started

You can download this dataset for use within your own projects, or fork it into a workspace on Roboflow to create your own model.

Dataset Versions

Version 1 (v1) - 496 images

  • Preprocessing: Auto-Orient and Resize (Stretch to 416x416)
  • Augmentations: No augmentations applied
  • Training Metrics: This version of the dataset was not trained

Version 2 (v2) - 1,190 images

  • Preprocessing: Auto-Orient, Resize (Stretch to 416x416), all classes remapped (Modify Classes) to "pill"

  • Augmentations:
    Outputs per training example: 3
    90° Rotate: Clockwise, Counter-Clockwise, Upside Down
    Shear: ±5° Horizontal, ±5° Vertical
    Hue: Between -25° and +25°
    Saturation: Between -10% and +10%
    Brightness: Between -10% and +10%
    Exposure: Between -10% and +10%
    Noise: Up to 2% of pixels
    Cutout: 5 boxes with 5% size each

  • Trained from the COCO Checkpoint in Public Models ("transfer learning") on Roboflow

NOTE:

The Isolate Objects preprocessing step was added to convert the original object detection project into a suitable format for export in OpenAI's CLIP annotation format so that it could be used as a classifcation model in this project.

Mohamed Attia - LinkedIn

Maintainer

mohamed-attia-e2mor

Last Updated

5 months ago

Project Type

Classification

Subject

pills

Classes

pill

License

CC BY 4.0