minseo0214/image-classification-level1-21

image-classification-level1-21 created by GitHub Classroom

TEAM 21 "BLACKJACK"

• "TEAM 21 Notion"

• TEAM 21 "BLACKJACK"

  • Image Classification Leaderboard
  • Model Visualization
    • ResNet50
    • EfficientNet b2
  • Usage
    • Train
      • Argument
      • Using Argument
    • Inference
      • Argument
      • Using Argument

Model Visualization

ResNet50

EfficientNet b2

Usage

Train

Argument

( '--model'      , type=str   , default='Customresnet50'    )
( '--dataset'    , type=str   , default='MaskBaseDataset'   )
( '--data_dir'   , type=str   )

Using argument

$ python3 train.py --model Customresnet50 --dataset SplitByProfileDataset --data_dir /your/data/dir 

Inference

Argument

( '--model'      , type=str   , default='EfficientResnet'   )
( '--data_dir'   , type=str   )
( '--model_dir'  , type=str   )

Using argument

$ python3 inference.py --model EfficientResnet --model_dir your/model/path.pth --data_dir /your/data/dir