Pill Classification Image Dataset
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.
The dataset is available under the Public License.
You can download this dataset for use within your own projects, or fork it into a workspace on Roboflow to create your own model.
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"
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
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