pytorch_img_classification_for_competition:使用pytorch进行图像分类 源码
pytorch图像分类竞赛框架 1.更新日志 (2020年5月2日)基础版本上线 2.依赖库 预训练模型 进步 高效网络火炬 顶尖 3.支持功能 pytorch官网模型 复现的部分模型 fp16混合精度训练 贸易协定 固定验证集/随机划分验证集 多种优化器:adam,radam,novograd,sgd,ranger,ralamb,over9000,lookahead,lamb OneCycle训练策略 标签平滑丢失 焦点损失 听觉 4.使用方法 更改config.py中的参数,训练执行python main.py ,预测执行python test.py 5. Submit_example.csv 每一行:filename,label样例: 0001.jpg,dog 0002.jpg,dog 0003.jpg,dog 注:预测图像可能没有标签,所以标签可以随意给个临时的
用户评论