行人搜索
FeboReigns
这个作者很懒,什么都没留下…
展开
-
yolov5自定义剪枝层 预测头剪枝
剪枝完成后会保存每一层的输入通道数和输出通道数,因此需要定义一个类接受这些参数。即使没对主干网络剪枝为了代码规范统一也要定义新的MobileNet_Block。具体的,主要在原来MobileNet_Block的基础上加入input和output参数,有多少卷积层就加多少组。原创 2023-03-20 18:25:55 · 846 阅读 · 12 评论 -
Reid Strong Baseline 代码解析
因此我需要frame id 这个属性,测试时把不同frame id 的全部图像作为gallery,也就是除去他本身外的所有图像。(1)新建一个类Market1501_ignore_cam,类似于market1501.py中的代码,只不过改下正则化规则。self 是自己的模型,类型为self时会接续训练,只需把start设置为0,则可以当作第0个epoch。1)修改数据集的路径。红色部分是之前的,绿色部分是修改后的。下面是正则化规则,也就是如何根据文件名读取标签,输出为文件路径,人物id,相机id。原创 2023-03-19 23:53:09 · 972 阅读 · 11 评论 -
基于分离框架的行人搜索所测试方法(以PRW为例)
主要是根据IoU给检测结果分配id,并保存。如未检测到目标,则保存一个纯黑的图片。下面的代码是根据检测结果生成reid的测试集(不包括 -1 -2标签的人物)数据格式为类别和box(中心点xy 和宽高 单位1为原图大小宽高)用YOLOv5检测行人,得到的结果为 图片名去掉后缀加.txt。如图像为检测到目标则不会有txt。原创 2023-01-11 00:08:52 · 232 阅读 · 0 评论 -
AlignPS test_result_prw.py注释
def main(det_thresh=0.05, gallery_size=-1, ignore_cam_id=True, input_path=None): #results_path = '/raid/ljp/code/chao_mmdetection/jobs/dcn_base_focal/' # change here results_path = '/home/yy1/2021/mmdetection-public/work_dirs/' + input_path .原创 2022-04-06 20:54:09 · 495 阅读 · 0 评论 -
CUHK 和PRW转为YOLO标签
行人搜索中有ID 标签(第二列),但是目标检测中没有ID,因此要把第二列去掉。xxxxxxxxxxxxxxxxxxx原创 2022-08-16 00:23:06 · 777 阅读 · 0 评论 -
PRW 数据集mat 转txt
PRW 有两个版本,有一个分了测试集和训练集但是标签是mat 类型,还有一个标签是txt 类型但是没分测试集和训练集。为了对比效果,我把原始的mat 类型的变成了txt 类型的。这是PRW的目录结构,frame_test.mat 存放的是所有测试集的文件名,annotations存放的是所有的图片的标签包括训练集和测试集。最后的目录结构张这样。原创 2022-10-10 21:38:15 · 355 阅读 · 0 评论