Tasfiya025
/

image-classifier-simple

Model card Files Files and versions

image-classifier-simple / README.md

Tasfiya025's picture

Create README.md

b427b8b verified about 1 month ago

|

history blame contribute delete

225 Bytes

Image classifier simple

Put images into data/train/ and data/val/
pip install -r requirements.txt
Train: python train_infer.py --mode train
Predict: python train_infer.py --mode predict --image path.jpg