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目录下即可

相关推荐
Ven%2 小时前
如何让后台运行llamafactory-cli webui 即使关掉了ssh远程连接 也在运行
运维·人工智能·chrome·python·ssh·aigc
Jeo_dmy2 小时前
(七)人工智能进阶之人脸识别:从刷脸支付到智能安防的奥秘,小白都可以入手的MTCNN+Arcface网络
人工智能·计算机视觉·人脸识别·猪脸识别
睡觉狂魔er4 小时前
自动驾驶控制与规划——Project 5: Lattice Planner
人工智能·机器学习·自动驾驶
xm一点不soso4 小时前
ROS2+OpenCV综合应用--11. AprilTag标签码跟随
人工智能·opencv·计算机视觉
云卓SKYDROID5 小时前
无人机+Ai应用场景!
人工智能·无人机·科普·高科技·云卓科技
是十一月末5 小时前
机器学习之过采样和下采样调整不均衡样本的逻辑回归模型
人工智能·python·算法·机器学习·逻辑回归
小禾家的5 小时前
.NET AI 开发人员库 --AI Dev Gallery简单示例--问答机器人
人工智能·c#·.net
生信碱移5 小时前
万字长文:机器学习的数学基础(易读)
大数据·人工智能·深度学习·线性代数·算法·数学建模·数据分析
KeyPan5 小时前
【机器学习:四、多输入变量的回归问题】
人工智能·数码相机·算法·机器学习·计算机视觉·数据挖掘·回归
码力全開5 小时前
C 语言奇幻之旅 - 第14篇:C 语言高级主题
服务器·c语言·开发语言·人工智能·算法