View Code? Open in Web Editor
NEW
p1-img-sooyounlee created by GitHub Classroom
Python 88.05%
Jupyter Notebook 11.95%
p1-img-sooyounlee's Introduction
pstage_02_image_classification
- torch==1.6.0
- torchvision==0.7.0
pip install -r requirements.txt
SM_CHANNEL_TRAIN=[train image dir] SM_MODEL_DIR=[model saving dir] python train.py
SM_CHANNEL_EVAL=[eval image dir] SM_CHANNEL_MODEL=[model saved dir] SM_OUTPUT_DATA_DIR=[inference output dir] python inference.py
SM_GROUND_TRUTH_DIR=[GT dir] SM_OUTPUT_DATA_DIR=[inference output dir] python evaluation.py
p1-img-sooyounlee's People
Contributors