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

最近更新

  1. 【软件工程中的演化模型及其优缺点】

    2023-12-13 21:10:04       0 阅读
  2. Linux 例题及详解

    2023-12-13 21:10:04       0 阅读
  3. 金蝶云苍穹-插件开发(四)GPT开发相关插件

    2023-12-13 21:10:04       1 阅读
  4. Windows图形界面(GUI)-SDK-C/C++ - 应用程序结构

    2023-12-13 21:10:04       1 阅读
  5. MySQL 速记SQL语句(SQL语句学习)

    2023-12-13 21:10:04       1 阅读
  6. xformers版本与其依赖pytorch版本的对应关系

    2023-12-13 21:10:04       1 阅读

热门阅读

  1. 【无标题】Vue3想在scss中想使用动态的变量

    2023-12-13 21:10:04       29 阅读
  2. 鸿蒙OS应用开发之最简单的程序

    2023-12-13 21:10:04       33 阅读
  3. [CF704E] Iron man

    2023-12-13 21:10:04       42 阅读
  4. C++异步网络库workflow入门教程(1)HTTP任务

    2023-12-13 21:10:04       45 阅读
  5. 自媒体短视频二次创作,收藏级干货

    2023-12-13 21:10:04       38 阅读
  6. Python 课堂记录

    2023-12-13 21:10:04       25 阅读
  7. 【Electron】富文本编辑器之回车换行

    2023-12-13 21:10:04       37 阅读
  8. HarmonyOS应用开发者基础认证【满分答案】

    2023-12-13 21:10:04       39 阅读