1. 首页
  2. 数据库
  3. 其它
  4. pytorch_classification:利用pytorch实现图像分类的一个完整的代码训练预测TTA模型融合模型部署cnn提取特征svm或者随机森林等进行

pytorch_classification:利用pytorch实现图像分类的一个完整的代码训练预测TTA模型融合模型部署cnn提取特征svm或者随机森林等进行

上传者: 2021-02-09 15:13:09上传 ZIP文件 4.39MB 热度 7次
pytorch_classification 利用pytorch实现图像分类,其中包含的密集网,resnext,mobilenet,efficiencynet,resnet等图像分类网络,可以根据需要再行利用torchvision扩展其他的分类算法 实现功能 基础功能利用pytorch实现图像分类 包含带有warmup的cosine学习率调整 warmup的step学习率优调整 多模型融合预测,修正与投票融合 利用flask实现模型云端api部署 使用tta测试时增强进行预测 添加label smooth的pytorch实现(标签平滑) 添加使用cnn提取特征,并使用SVM,RF,MLP,KN
下载地址
用户评论