File size: 225 Bytes
b427b8b |
1 2 3 4 5 6 |
Image classifier simple
- Put images into data/train/<class> and data/val/<class>
- pip install -r requirements.txt
- Train: python train_infer.py --mode train
- Predict: python train_infer.py --mode predict --image path.jpg
|