mxer/ImageClassificationTemplate

This is my image classification template using pytorch, support opencv and PIL

ImageClassificationTemplate

This is an image classification finetune template using torchvision or timm.

1.prerequisites

python3.7+、gpu

2.installation

pip install -r requirements.txt

3.Support List

• Multi-GPU Training Solution

  • DistributedDataParallel (DDP) [Built-in function of Pytorch]

• Pretrain Model

  • totchvision
  • timm
  • local

• Deploy

  • onnx(support simplify and optimize)
  • mnn

• Log

  • logger
  • tensorboard