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 注:预测图像可能没有标签,所以标签可以随意给个临时的
文件列表
pytorch_img_classification_for_competition-master.zip
(预估有个23文件)
pytorch_img_classification_for_competition-master
models
__init__.py
20B
model.py
3KB
ensemble.py
1KB
main.py
12KB
test.py
5KB
LICENSE
11KB
.gitignore
2KB
config.py
2KB
暂无评论