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