【文件属性】:
文件名称:x-classification:基于pytorch的图像分类框架
文件大小:68KB
文件格式:ZIP
更新时间:2021-05-29 11:03:57
Python
pytorch 分类
使用 Pytorch 在CIFAR10 CIFAR100和ImageNet进行分类
特征
两种训练方式
纪元训练:使用固定纪元的训练网络(warmup_epoch/epoch=1/200)
step training:使用固定步长训练网络(warmup_steps/total_steps= 1000/80000)
多 GPU 支持
简单实用的培训日志文件
支持不同的训练计划( multi-step / cosine / linear )
要求
Python3.6
pytorch1.6.0 + cuda10.1
张量板 2.3.0
翻牌
安装
克隆 git clone https://github.com/x-multimodal/x-classification.git
用法
1.进入目录
$ cd x-classification
2. 数据集
现在仅支持
【文件预览】:
x-classification-master
----.gitignore(37B)
----step_run.py(3KB)
----datasets()
--------dataset()
--------__pycache__()
--------dataset_loader.py(3KB)
----criterion()
--------__init__.py(53B)
--------cross_entropy.py(3KB)
----models()
--------get_network.py(799B)
--------net()
--------__pycache__()
----utils()
--------scheduler.py(3KB)
--------DALI.py(0B)
--------testing()
--------util.py(4KB)
--------training()
--------visual_engine.py(0B)
----epoch_run.py(3KB)
----README.md(7KB)
----.idea()
--------.gitignore(222B)
--------misc.xml(195B)
--------MAC_classification.iml(284B)
--------inspectionProfiles()
--------modules.xml(288B)
----net_stat.py(951B)
----counter.py(1KB)
----conf()
--------model_configs()
--------__init__.py(2KB)
--------data_configs()
--------global_configs.py(738B)
--------__pycache__()