pstage_01_image_classification

Getting Started

Dependencies

• torch==1.7.1
• torchvision==0.8.2

Install Requirements

• pip install -r requirements.txt

Training

• Example : EfficientNet_b0/config.yml 설정 값으로 학습
  • python -m train --config_file=configs/EfficientNet_b0/config.yml

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