qudgns5129 / pill_classification

Broken Pill Classification CNN Model

Broken Pill Classification

---

Broken and Undroken pill images Classification

Sequence

---

1. Load image link from data.csv and save it to File system
2. image preprocessing -> image segmentation
3. Manually omitting some of the pills
4. Classification with CNN Models

Example Data

---

Raw image

Image with segmentation

Saved image

Making Broken Pill images

Models

---

• VGG16
• GoogLeNet
• ResNet50