cuteboyqq / Classify-Pytorch

classify model

Classify-model

(Back to top)

in folder network

1. resnet
2. repVGG
3. res2net
4. VGG

Demo-Preview

(Back to top)

Table of contents

• Classify-model
• Demo-Preview
• Usage
  • parameter-setting
  • Train
  • Test

Usage

(Back to top)

parameter-setting

(Back to top)

parameter are set in yaml/train_cfg.yaml and yaml/inference_cfg.yaml

Train

(Back to top)

python train.py 

Test

(Back to top)

python test.py