image-classification-models:PyTorch; 图像分类; CNN模型; CIFAR10

时间:2021-04-02 10:07:42
【文件属性】:
文件名称: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)

网友评论