0 前言
前面开源了行人检索的代码。由于是多数据集联合训练,因此泛化效果还是不错的,一些简单场景都可以直接用。具体可以看看下面,可爱的实验室师兄师姐们拍摄的演示视频进行测试的结果。
下面具体介绍一下,这个演示视频是具体怎么进行测试的:
1. 截取得到待检索的行人图片
对于侦探寻找犯人来说,往往是有一个中心现场的监控得到了犯罪嫌疑人的图片。因此,我们先利用 query_get.py 截取中心现场视频中的行人图片。
修改后路径之后,就开始运行了,点击空格快速定位到自己觉得带查找行人特征比较明显的地方就停下来:
点击鼠标进行拖拽,完整的框住查找的行人:
可以看到,图片自动被保存到了带查找行人库query中,这时候需要手动剔除截取的不好的或者非常相似的:
2. 放置好待查找图片或视频
将需要进行查找的图片或者视频放置在 person_search_demo/data/samples 中即可
这样数据部分就大功告成了。
3. 运行 search.py进行查找
程序运行结束后,处理好的图片和视频都会存放在output文件夹中,可以进行查看了
注意点:
- dist_thres 阈值可能有时候需要根据自己的具体情况进行一些调整
- query的查找行人图片数目越多,速度也会越慢一些
- 对于只需要快速查找行人的视频场景,可以采取更多的跳帧