利用pytorch实现图像分类的一个完整的代码训练预测TTA模型融合模型部署cnn提取特征svm或者随机森林等进行
利用pytorch实现图像分类的一个完整的代码训练预测TTA模型融合模型部署cnn提取特征svm或者随机森林等进行分类模型蒸馏一个完整的代码主要功能利用pytorch实现图像分类基于torchision可以扩展使用densenet resnext mobilenet efficientnet swin transformer等图像分类网络
文件列表
pytorch_classification-master.rar
(预估有个53文件)
pytorch_classification-master
tools
evaluation.py
5KB
train_val.py
6KB
predict.py
2KB
sample_files
imgs
dog
2.jpg
15KB
1.jpg
26KB
暂无评论