license: agpl-3.0
language:
- en
metrics:
- accuracy
library_name: transformers
tags:
- medical
pipeline_tag: image-classification
This model is part of a school project. Utilizing the google/vit-base-patch16-224 vision transformer for image classification, this pre-trained model is further tuned utilizing images of pills and tablets. As pills and tablets have three main features, color, shape, and imprint, the model aims to identify images of pill and tablets by automatically extracting features. The dataset utilized is from the U.S. Department of Health's Computational Photography Project for Pill Identification (C3PI).
DISCLAIMER: The accuracy of this model is currently low (<20%). Further training is currently ongoing to improve the accuracy.
Version 2: This version tries to train the pretrained model with only 20 of the most common pills. Unfortunately, the accuract of the model is still currently low (<30%).