文件名称:pytorch_img_classification_for_competition:使用pytorch进行图像分类
文件大小:29KB
文件格式:ZIP
更新时间:2024-04-09 15:04:48
competition pytorch image-classification Python
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
----models()
--------__init__.py(20B)
--------model.py(3KB)
----ensemble.py(1KB)
----main.py(12KB)
----test.py(5KB)
----LICENSE(11KB)
----.gitignore(2KB)
----config.py(2KB)
----README.md(1KB)
----utils()
--------optimizers()
--------logger.py(4KB)
--------losses()
--------misc.py(4KB)
--------__init__.py(69B)
--------reader.py(1KB)