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

This topic created in 1553 days ago, the information mentioned may be changed or developed.

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

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

应该如何训练？

提供大量网络头像吗？

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

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

3 replies • 2022-08-15 10:48:47 +08:00

Juszoe

Aug 14, 2022

“识别聊天窗口的头像位置”感觉用 OpenCV 足够了，yolo 有些杀鸡用牛刀，识别出区域后再裁剪图片分类。
如果你一定要 yolo ，可以把目标分为三类，猫、狗、其他

Muniesa

Aug 14, 2022 via Android

你需要聊天窗口的图片，标注出里面的头像位置，至于类别标签，直接给个头像类就行了，不需要管头像内容。当然你的数据集里头像应当是包含各种图片的。

EeffDev

Aug 15, 2022

@Juszoe 我试了一下 OpenCV ，边缘查找，尝试了很多参数，很多情况下都会把头像给破除开，就是一个淡色背景的头像，会直接分离出来头像内的轮廓。最终还是没办法识别