Commit History
-
Create README.md
Verified signature65bcc7e -
README.md
Verified signature# Image Classifier (Simple PyTorch) ## Install ``` pip install torch torchvision Pillow ``` ## Dataset Format ``` data/train/class1/*.jpg data/train/class2/*.jpg data/val/class1/*.jpg data/val/class2/*.jpg ``` ## Train ``` python train_infer.py --mode train --epochs 3 ``` ## Predict ``` python train_infer.py --mode predict --image test.jpg ```6b76db3 -
initial commit
Verified signature0d71110