使用 yolo 训练识别头像区域的思路应该是怎样的?

2022-08-14 14:48:00 +08:00
 EeffDev

识别猫狗的数据集,只要有猫和狗的图片就好了

如果要识别聊天窗口的头像位置

应该如何训练?

提供大量网络头像吗?

但是头像可能是猫,可能是狗,也可能是人,或者什么也不是

这样应该如何提供训练集呢?

956 次点击
所在节点    问与答
3 条回复
Juszoe
2022-08-14 16:33:11 +08:00
“识别聊天窗口的头像位置”感觉用 OpenCV 足够了,yolo 有些杀鸡用牛刀,识别出区域后再裁剪图片分类。
如果你一定要 yolo ,可以把目标分为三类,猫、狗、其他
Muniesa
2022-08-14 17:39:47 +08:00
你需要聊天窗口的图片,标注出里面的头像位置,至于类别标签,直接给个头像类就行了,不需要管头像内容。当然你的数据集里头像应当是包含各种图片的。
EeffDev
2022-08-15 10:48:47 +08:00
@Juszoe 我试了一下 OpenCV ,边缘查找,尝试了很多参数,很多情况下都会把头像给破除开,就是一个淡色背景的头像,会直接分离出来头像内的轮廓。最终还是没办法识别

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/872765

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX