HRNet训练自己的分类数据

本文主要是介绍HRNet训练自己的分类数据，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

目录

1.搭建环境

1.1创建conda环境

1.2下载HRNet工程

1.3安装依赖

2.准备数据集

3.修改配置文件

4.增加保存pth模型的代码

5.增加代码保存分类名称和索引

6.开始训练

7.测试

---

---

1.搭建环境

1.1创建conda环境

为了防止和服务器上的环境冲突，这里利用conda搭建环境。

conda create -n HRNet_chw python=3.7
conda activate HRNet_chw

1.2下载HRNet工程

git clone https://github.com/HRNet/HRNet-Image-Classification

1.3安装依赖

首先修改工程里面的requirement.txt。其中增加torchvision==0.8.2和torch==1.7.1。另外由于工程里面原有的requirement.txt文件里面某些依赖的版本找不到，所以把一些版本删掉，修改后的requirement.txt如下：

torchvision==0.8.2
torch==1.7.1
EasyDict==

这篇关于HRNet训练自己的分类数据的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---