pytorch_classification
利用pytorch实现图像分类,其中包含的密集网,resnext,mobilenet,efficiencynet,resnet等图像分类网络,可以根据需要再行利用torchvision扩展其他的分类算法
实现功能
基础功能利用pytorch实现图像分类
包含带有warmup的cosine学习率调整
warmup的step学习率优调整
多模型融合预测,修正与投票融合
利用flask实现模型云端api部署
使用tta测试时增强进行预测
添加label smooth的pytorch实现(标签平滑)
添加使用cnn提取特征,并使用SVM,RF,MLP,KN
1