songwsx / person_search_demo Goto Github PK
View Code? Open in Web Editor NEW利用YOLOv3结合行人重识别模型,实现行人的检测识别,查找特定行人
License: MIT License
利用YOLOv3结合行人重识别模型,实现行人的检测识别,查找特定行人
License: MIT License
你好我的笔记本没有GPU但是我想测试您的代码,在运行python search.py时报错如下:
$ python search.py
Namespace(cfg='cfg/yolov3.cfg', conf_thres=0.1, data='data/coco.data', dist_thres=1.0, fourcc='mp4v', half=False, images='data/samples', img_size=416, nms_thres=0.4, output='output', query='query', webcam=False, weights='weights/yolov3.weights')
Using CPU
Traceback (most recent call last):
File "search.py", line 226, in
output=opt.output)
File "search.py", line 38, in detect
reidModel.load_param(reidCfg.TEST.WEIGHT)
File "D:\person_search_demo-master\reid\modeling\baseline.py", line 63, in load_param
param_dict = torch.load(trained_path)
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\serialization.py", line 386, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\serialization.py", line 573, in _load
result = unpickler.load()
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\serialization.py", line 536, in persistent_load
deserialized_objects[root_key] = restore_location(obj, location)
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\serialization.py", line 119, in default_restore_location
result = fn(storage, location)
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\serialization.py", line 95, in _cuda_deserialize
device = validate_cuda_device(location)
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\serialization.py", line 79, in validate_cuda_device
raise RuntimeError('Attempting to deserialize object on a CUDA '
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
该怎么解决,新手求教!不胜感激!
您好!
有个问题请教一下。您展示的结果,“白衣服女生在C楼前走过”,以及“白帽子男生在校门前走过",这两个例子处理的都是同一摄像头在相近的时间内拍下的帧对吧?如果是非同一摄像头拍下的帧,用这个模型处理,是不是就效果不佳了?
在“search.py”中,请问下,dist_thres是如何计算得到的?
Loading pretrained ImageNet model......
torch.Size([128, 2048])
The query feature is normalized
torch.Size([128, 2048])
Traceback (most recent call last):
File "search.py", line 259, in
detect(opt.cfg,
File "search.py", line 87, in detect
query_feats.append(feat)
AttributeError: 'Tensor' object has no attribute 'append'
作者您好,我使用自己在 reid-strong-baseline 训练得到的模型,运行search.py时报错,报错信息:
Traceback (most recent call last):
File "E:\person_search_demo\search.py", line 228, in
output=opt.output)
File "E:\person_search_demo\search.py", line 38, in detect
reidModel.load_param(reidCfg.TEST.WEIGHT)
File "E:\person_search_demo\reid\modeling\baseline.py", line 63, in load_param
param_dict = torch.load(trained_path)
File "d:\anaconda3\envs\torch\lib\site-packages\torch\serialization.py", line 387, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "d:\anaconda3\envs\torch\lib\site-packages\torch\serialization.py", line 574, in _load
result = unpickler.load()
ModuleNotFoundError: No module named 'modeling'
请问可以提供一些解决的建议吗?
松老师是不是这个代码有个测试行人的时长限制啊,,我查到末尾就找不出来了
作者你好,非常感谢你的开源分享,已经star。因为刚开始自己进行这方面的训练测试等还不熟悉,你的博客里面写的很清楚,但是能不能分享一下完整代码,想看一下检测和行人识别代码中如何相结合的,以及整个网络的架构,谢谢了。抱拳
松老师您好,发现您做的 person_search_demo 很不错,谢谢您愿意与大家分享!
有注意到您没特别注明 license,不知您可否在 GitHub 补上常见的开源 license (例如 MIT) 让大家放心使用?
请问这个是什么公式?用来将距离转成相似度的吗? @songwsx
送入reid网络的gallery图片是全景图片还是单个人抠出来的图片
ModuleNotFoundError: No module named 'reidCfg'我在运行search的时候报这个错误是什么原因啊
运行search.py遇到这个bug,需要怎么解决呢
如上
我data/samples里面放的自己的视频帧
query里面放的截取出来的人体
然后我运行程序python3 search.py报错
我git下来,运行你的例子没问题,我把图片这些换成我的就不行了,求解答,谢谢作者
@songwsx
I make a scripts of evaluation on PRW dataset, the result is good, do you need it?
You can use it directly, and I have test your demo, the result is good.
作者你好,非常感谢你的开源分享,已经star, 想问yolov3-spp权重文件,如果想重新训练,你采用的是那个版本呢 ?pytorch的吗?
RT
请问怎么把分类数据转为market1501格式,分类数据:一个文件夹包含一个类别的图片
您好,首先非常感谢您愿意分享您的代码
但是我在运行python search.py时报错,报错为No module named 'yacs'
我看了下代码在search.py,import语句中reid.config中确实没有cfg这个文件,在其他文件中也没有。请问一下这个文件在哪个位置。谢谢您了
行人重识别数据集链接失效了,请问可以重新更新一下吗?
请问怎么修改检测框上面的文字?
提示缺少weights/719rank1.pth,请问这个文件需要在哪里找到。
A declarative, efficient, and flexible JavaScript library for building user interfaces.
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google ❤️ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.