/level1_imageclassification_cv-level1-cv-07

level1_imageclassification_cv-level1-cv-07 created by GitHub Classroom

Primary LanguagePython

pstage_01_image_classification

Getting Started

Dependencies

  • torch==1.7.1
  • torchvision==0.8.2

Install Requirements

  • pip install -r requirements.txt

Training

  • SM_CHANNEL_TRAIN={YOUR_TRAIN_IMG_DIR} SM_MODEL_DIR={YOUR_MODEL_SAVING_DIR} python train.py

Inference

  • SM_CHANNEL_EVAL={YOUR_EVAL_DIR} SM_CHANNEL_MODEL={YOUR_TRAINED_MODEL_DIR} SM_OUTPUT_DATA_DIR={YOUR_INFERENCE_OUTPUT_DIR} python inference.py

Evaluation

  • SM_GROUND_TRUTH_DIR={YOUR_GT_DIR} SM_OUTPUT_DATA_DIR={YOUR_INFERENCE_OUTPUT_DIR} python evaluation.py