马春杰杰博客
致力于深度学习经验分享!

[mcj]修改YOLO使其显示自定义类别

文章目录
[隐藏]

根据@赤战约风 的帖子做如下修改可以使其只显示人这个类别

一.修改源代码

cfg/coco.data

src/detector.c

二.重新编译和测试

1.重新编译

2.测试

这样的修改其实还是有点麻烦的,参考yangshun8889大神的回答,其实还可以这样:

我假设你使用的是VOC.name, 在VOC中,car对应着6, person对应着14,然后修改代码。在draw_detections()函数中第一个for循环int class = max_index(probs[i], classes);代码后面添加if(class!=6 && class!=14){continue;}

这样就可以检测人和车了。

参考链接:

http://blog.csdn.net/u012235003/article/details/54576974

https://groups.google.com/d/msg/darknet/9Geu1zebfxo/rt2zLVpVAwAJ

本文最后更新于2019年6月13日,已超过 1 年没有更新,如果文章内容或图片资源失效,请留言反馈,我们会及时处理,谢谢!

如果你对这篇文章有什么疑问或建议,欢迎下面留言提出,我看到会立刻回复!

打赏
未经允许不得转载:马春杰杰 » [mcj]修改YOLO使其显示自定义类别

留个评论吧~ 抢沙发 评论前登陆可免验证码!

私密评论

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址(选填,便于回访^_^)
切换注册

登录

忘记密码 ?

切换登录

注册