关于reid-strong-baseline的一些小坑

打算水一篇

今天需要跑一下reid-strong-baseline这个项目,该项目提供了很多训练脚本,然后,他提供支持了market1501、dukemtmc等数据集的读取代码,这些数据集结构都非常简单,因为我要用我自己的数据,然后自己写这些代码略麻烦,就改一下我的数据格式,无非就是改改图片名字。

以dukemtmc为例,首先是必须要在传入的root路径下,有个名为dukemtmc-reid目录,里面必须包含有一个名为DukeMTMC-reID的目录,再在这个目录下有三个目录,分别为bounding_box_train,bounding_box_test和query,其中train里面是训练图片,test就是测试图片,query里面,按照介绍说是测试图片里面每个分类中的一张,但是直接取会报错,他是,文件名中包含了三个部分,以‘_’隔开,分别是人物ID(类别id),摄像头ID和第几帧,这里面只有类别id对我的数据是有意义的,摄像头,我就都是填的‘c1’,帧数就是随机数了,这里面我的query是从test中每个类别抽一张。结果在训练中的eva的时候,报错,似乎是因为他会将query里面出现在test中的同类别与同摄像头的去掉,然后这样的话query就没有图片了,最终的解决是,在query里面,把每个图片的摄像头id改一下,这样就可以了。此外,由于我的数据和数据集的不一样,在读取图像的时候,他normlize肯定是不一样的哦,我直接去掉那一项了,并且,他是直接将图片resize的,所以需求不一样的话,这部分要好好处理一下了。

在训练之前,会提示缺少库,比较坑的是他会提示缺少ignite,于是pip安装之后,很顺利,但是再运行,会报缺少ignite.engine,这就!!!经过这篇的讲解,发现原来是需要安装pytorch-ignite,你说这谁想得到。但是装完还是会报错,查git的issues,发现,对版本有要求,需要安装0.1.2的版本。

其他的似乎没有什么了,成功水一篇。

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值