Skip to content

Latest commit

 

History

History
13 lines (10 loc) · 549 Bytes

README.md

File metadata and controls

13 lines (10 loc) · 549 Bytes

2D Image Classification

Requirements

  • Python 3.8
  • PyTorch
  • Torchvision

Train and Evaluate

  • cd ".\ResNetImageClassification"
  • python train/main.py --image_dir dataset --model_dir models --epochs 10 - Train the model
  • python train/main.py --image_dir dataset --model_dir models --warmup_model_dir models --epochs 10 - Train the model from a checkpoint
  • python eval_image.py --image_dir ./run --model_dir ./models --output ./predictions.txt - To evaluate images. See the output of python eval_image.py --help for arguments.