Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Image classifier simple
|
| 2 |
+
- Put images into data/train/<class> and data/val/<class>
|
| 3 |
+
- pip install -r requirements.txt
|
| 4 |
+
- Train: python train_infer.py --mode train
|
| 5 |
+
- Predict: python train_infer.py --mode predict --image path.jpg
|