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

时间:2024-04-22 03:54:22
【文件属性】:

文件名称:image-classification-models:PyTorch; 图像分类; CNN模型; CIFAR10

文件大小:5KB

文件格式:ZIP

更新时间:2024-04-22 03:54:22

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)

网友评论