Instructions to use ph0masta/bug_classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ph0masta/bug_classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="ph0masta/bug_classifier") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("ph0masta/bug_classifier") model = AutoModelForImageClassification.from_pretrained("ph0masta/bug_classifier") - Notebooks
- Google Colab
- Kaggle
Excuse me, but can you share the dataset you used for training?
#2
by ckw1140 - opened
This comment has been hidden
Sorry but I deleted the images. I believe I found them on Wikipedia Commons
There are images in the source repo but I believe those are just examples
This comment has been hidden
Sorry but I don’t remember if there was any pre-processing beyond downsizing the images. I believe there was typically 100-300 per class.
ckw1140 changed discussion status to closed