Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
zonghuia
04c14cd9c6
|
1 year ago | |
|---|---|---|
| README.md | 1 year ago | |
| load_data.py | 1 year ago | |
| main.py | 1 year ago | |
| model.py | 1 year ago | |
| tools.py | 1 year ago | |
| train.ipynb | 1 year ago | |
| utils.py | 1 year ago | |
README.md
Project Introduction
这个项目是一个图像分类任务，数据集使用 CIFAR10 数据集，共有十个类别，训练集中每个类别下有 5000 张图片，测试集中每个类别下有 1000 张图片。
模型使用 ResNet50 作为分类任务模型
各个文件介绍
load_data.py 加载数据集的程序文件，里边包含了加载数据集的类
main.py 训练程序文件，程序的入口，并且在里边配置训练参数
model.py 2D ResNet 模型文件，包含了残差网络模型，并且可选残差网络的深度
tools.py 根据数据集生成包含数据集路径和标签的 txt 文件
train.ipynb 测试程序的 jupyter 文件
utils.py 包含准确率和召回率的计算程序