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

相关推荐
2501_9418372620 分钟前
多颜色玫瑰品种识别与分类_YOLO13-C3k2-PoolingFormer模型详解_1
人工智能·数据挖掘
Loacnasfhia930 分钟前
面部表情识别与分类_YOLOv10n与MobileNetV4融合方案详解
yolo·分类·数据挖掘
新缸中之脑41 分钟前
为什么我选 Codex
人工智能
yumgpkpm44 分钟前
2026软件:白嫖,开源,外包,招标,晚进场(2025年下半年),数科,AI...中国的企业软件产业出路
大数据·人工智能·hadoop·算法·kafka·开源·cloudera
witAI1 小时前
**AI漫剧制作工具2025推荐,零成本实现专业级动画创作*
人工智能·python
冬奇Lab1 小时前
一天一个开源项目(第12篇):SoulX-Podcast - 多轮对话式播客生成,让AI语音更自然真实
人工智能·开源
风栖柳白杨1 小时前
【语音识别】一些音频的使用方法
人工智能·音视频·语音识别
xixixi777771 小时前
今日 AI 、通信、安全行业前沿日报(2026 年 2 月 4 日,星期三)
大数据·人工智能·安全·ai·大模型·通信·卫星通信
LucDelton1 小时前
模型微调思路
人工智能·深度学习·机器学习
寻道码路1 小时前
【GitHub开源AI精选】WhisperX:70倍实时语音转录、革命性词级时间戳与多说话人分离技术
人工智能·开源·github