pytorch_img_classification_for_competition

pytorch_img_classification_for_competition copied to clipboard

pytorch 图像分类竞赛框架

1. 更新日志

• (2020年5月2日) 基础版本上线

2. 依赖库

• pretrainedmodels
• progress
• efficientnet-pytorch
• apex

3. 支持功能

• [x] pytorch官网模型
• [x] pretrained-models.pytorch 复现的部分模型
• [x] EfficientNet-PyTorch
• [x] fp16混合精度训练
• [x] TTA
• [x] 固定验证集/随机划分验证集
• [x] 多种优化器：adam、radam、novograd、sgd、ranger、ralamb、over9000、lookahead、lamb
• [x] OneCycle训练策略
• [x] LabelSmoothLoss
• [x] Focal Loss
• [ ] AotuAgument

4. 使用方法

更改config.py中的参数，训练执行 python main.py，预测执行python test.py

5. submit_example.csv

每一行：filename,label 样例：

0001.jpg,dog
0002.jpg,dog
0003.jpg,dog

注：预测图像可能没有label，所以label可以随意给个临时的，但一些比赛平台对都会给个提交样例，随意给个label

6.TODO

• [ ] 优化模型融合策略
• [ ] 优化online数据增强
• [ ] 优化pytorch官方模型调用接口
• [ ] 增加模型全连接层初始化
• [ ] 增加更多学习率衰减策略
• [ ] 增加find lr
• [ ] 增加dali
• [ ] 增加wsl模型
• [ ] 增加tensorboardX
• [ ] 优化文件夹创建