文件名称: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)