resnet 图像分类的项目

1. 项目文件

文件下载资源:resnet 图像分类的项目代码

本章利用reset34 作图像分类,包括计算训练集和测试集的loss、accuracy曲线,对训练好的网络在训练集+测试集上求混淆矩阵

  • data 文件为训练集+测试集,图像按照文件夹摆放
  • inference 负责放待推理的图片(支持多张图片预测分割)
  • run_results 是网络训练过程的日志文件
  • python 代码介绍。train 为训练函数,predict 为预测脚本,utils 为工具函数
  • class_indices.json 文件是代码生成的,不需要自行设置

2. 代码的使用

README 文件有介绍

按照指定位置摆放,文件夹的名称不可更改,否则需要重新编写dataset文件 !!!

将test和train目录下的文件夹更改,文件夹名称为分类标签

使用的时候只需要将test 和train 下面的目录换成自己的数据,相同的类别放在同一个文件夹下

超参数根据实际需要更改,以下为train.py 的超参数

必须更改的为num--classes,根据分类的个数更改!!也就是-data-train-下的文件夹个数

预测的时候,需要将下面参数更改!!

mean和std 会在train.py 运行的时候生成,copy即可

3. 展示结果

本章测试的数据集为5个花分类的任务

训练结果:

loss_accuracy_curve 为:

对训练集和测试集的混淆矩阵:

推理的时候,只需要将单个或者多个图像放在inference目录下即可

相关推荐
qq_12498707532 分钟前
基于JavaWeb的大学生房屋租赁系统(源码+论文+部署+安装)
java·数据库·人工智能·spring boot·计算机视觉·毕业设计·计算机毕业设计
杜子不疼.3 分钟前
CANN算子基础框架库opbase的算子开发与扩展机制深度解析
人工智能
程序猿追4 分钟前
CANN ops-math仓库解读 数学算子的底层支撑与高性能实现
人工智能·架构
结局无敌4 分钟前
统一算子语言:cann/ops-nn 如何为异构AI世界建立通用“方言”
人工智能·cann
杜子不疼.11 分钟前
CANN计算机视觉算子库ops-cv的图像处理与特征提取优化实践
图像处理·人工智能·计算机视觉
大闲在人12 分钟前
软件仍将存在,但软件公司会以全新形式出现——从Claude智能体引发万亿市值震荡看行业重构
人工智能
艾莉丝努力练剑12 分钟前
【Linux:文件】Ext系列文件系统(初阶)
大数据·linux·运维·服务器·c++·人工智能·算法
芷栀夏16 分钟前
从 CANN 开源项目看现代爬虫架构的演进:轻量、智能与统一
人工智能·爬虫·架构·开源·cann
梦帮科技26 分钟前
OpenClaw 桥接调用 Windows MCP:打造你的 AI 桌面自动化助手
人工智能·windows·自动化
User_芊芊君子29 分钟前
【分布式训练】CANN SHMEM跨设备内存通信库:构建高效多机多卡训练的关键组件
分布式·深度学习·神经网络·wpf