ImageClassificationTemplate

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

python3.7+、gpu

pip install -r requirements.txt

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

Name		Name	Last commit message	Last commit date
Latest commit History 39 Commits
data		data
loss		loss
models		models
utils		utils
README.md		README.md
__init__.py		__init__.py
finetune.py		finetune.py
inference.py		inference.py
inference_multi_gpu.py		inference_multi_gpu.py
inference_video.py		inference_video.py
mnn_inference.py		mnn_inference.py
onnx_test.py		onnx_test.py
predict_url.py		predict_url.py
requirements.txt		requirements.txt
test_multi_gpu.py		test_multi_gpu.py

Provide feedback