在自己的数据集上训练CrowdDet过程记录

2024-05-11 16:18:09

论文链接：https://readpaper.com/pdf-annotate/note?noteId=656650387498369024&pdfId=542662939605901312

github链接：https://github.com/Purkialo/CrowdDet

一：在CrowdDet上训练CrowdHuman数据集：

1.CrowdHuman数据集

①下载链接：CrowdHuman Dataset。

②数据集介绍：

其中训练集分为3部分，名字分别为CrowdHuman_train01.zip,CrowdHuman_train01.zip,CrowdHuman_train01.zip,数量都是5k张；测试集为CrowdHuman_test.zip,数量为5k,验证集为CrowdHuman_val.zip，数量是4370张。odgt文件则是标签，里面有全身标注，行人可见部分标注以及head部分标注。

2.训练CrowdHuman数据集

①：在config.py中，图上所标注的路径都是需要修改成自己项目中的路径，例如：我的"annotation_train.odgt"存放路径为"/home/mobu/CrowdHuman/annotation_train.odgt"，则train_source的路径需要改成这个，另外附上预训练模型链接：https://drive.google.com/file/d/1lfYQHC63oM2Dynbfj6uD7XnpDIaA5kNr/view?usp=sharing。

②：image_folder表示CrowdHuman图片存放的地方，前面下载的zip文件会解压到一个叫‘Images’的文件夹，所以上图标示部分应该是大写的‘Images’,另外路径也需要修改，参考第一点。

③：解压出来的图片后缀名都是'.jpg'，所以需要将'.png'修改为'.jpg'

一般来说，这几个地方修改之后就可以在CrowdHuman数据集上训练CrowdDet了。

二：自定义数据集训练

①我的数据集格式

如图所示，前面是图片路径，后面是一个列表中套字典的格式，每个字典中有左上角和右下角的标注以及目标类别。

②CrowdDet中的dataset

CrowdDet的dataset在lib/data/CrowdHuman.py，根据我的数据格式对相关地方做修改即可。

首先修改对数据的解析方式，返回一个标签列表：

parse_lines函数如下：

接下来修改图片路径以及对标签的解析方式：

parse_record函数如下：

需要注意的是，CrowdHuman数据中box列表中4个值的格式是[x0, y0, w, h]，即左上角的点和宽高，所以我们处理的时候也需要做转换。

最后再把验证阶段也修改一下就大功告成了！

在自己的数据集上训练CrowdDet过程记录相关推荐

internetreadfile读取数据长度为0_YOLOV3的TensorFlow2.0实现，支持在自己的数据集上训练...
GitHub链接: calmisential/YOLOv3_TensorFlow2github.com 我主要参考了yolov3的一个keras实现版本: qqwweee/keras-yolo3g ...
模拟数据集上训练神经网络，网络解决二分类问题练习
#2018-06-24 395218 June Sunday the 25 week, the 175 day SZ ''' 模拟数据集上训练神经网络,网络解决二分类问题.'''import tens ...
多目标跟踪算法JDE在 UA-DETRAC数据集上训练
环境:win10,cuda 10.1 , GTX1060 一.数据处理 1.数据集获取: 链接:https://pan.baidu.com/s/1K3rI9PvzHc1KqOJITNMdVg 提取码 ...
pascal行人voc_在一个很小的Pascal VOC数据集上训练一个实例分割模型
只使用1349张图像训练Mask-RCNN,有代码. 代码:https://github.com/kayoyin/tiny-inst-segmentation 介绍计算机视觉的进步带来了许多有前途的 ...
构建Transformer模型 | 在wikiText-2数据集上训练一个语言模型
0 Introduction 自然语言处理通用解决方案需要熟悉word2Vec, 了解词向量如何建模重点在于Transformer网络架构,BERT训练方法,实际应用开源项目,都是现成的,套用进 ...
DL之DCGNN：基于TF利用DCGAN实现在MNIST数据集上训练生成新样本
DL之DCGNN:基于TF利用DCGAN实现在MNIST数据集上训练生成新样本目录输出结果设计思路实现部分代码说明:所有图片文件丢失输出结果更新-- 设计思路更新-- 实现部分代码更 ...
domain gap(域间隙)是什么？==＞在一个数据集上训练好的模型无法应用在另一个数据集上
不同数据集之间存在domain gap,在一个数据集上训练模型,在另外一个数据集上进行预测性能下降很大 re-id(视频行人重识别问题) 现有公开的数据集与真实场景存在很大不同,不同re-id的数据集 ...
pascal行人voc_在Pascal VOC 数据集上训练YOLOv3模型
上节介绍了<从零开始在Windows10中编译安装YOLOv3>,本节介绍在Pascal VOC 数据集上训练YOLOv3. 第一步,下载并安装YOLOv3训练依赖项. a.下载Pasca ...
在MNIST数据集上训练一个手写数字识别模型
使用Pytorch在MNIST数据集上训练一个手写数字识别模型, 代码和参数文件可下载 1.1 数据下载 import torchvision as tvtraining_sets = tv.dat ...

最新文章

热门文章