A real-world image classification project using transfer learning with ResNet18.
You can train model with some image classes and predict what the input image is.
python train.pypython predict.py image.jpgnightmare0329/food-classifier
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|