【文件属性】:
文件名称:image-classification-models:PyTorch; 图像分类; CNN模型; CIFAR10
文件大小:5KB
文件格式:ZIP
更新时间:2021-04-02 10:07:42
Python
图像分类模型
PyTorch; 图像分类; CNN模型; CIFAR10
数据集
楷模
我们使用经典模型来训练图像分类。
ResNet
对于ResNet模型,您需要下载预模型并设置premodel_dir
premodel_dir = 'Here is the download premodel dir'
【文件预览】:
image-classification-models-main
----models()
--------__init__.py(1B)
--------resnet.py(15KB)
----utils()
--------__init__.py(1B)
--------train_utils.py(518B)
----README.md(310B)
----demo()
--------__init__.py(1B)