现在我识别一个 1080p 视频中的实心球,实心球会是各种颜色,但是一定是纯色,当然可能受天气影响,光照等,会有一些变化。实心球大概 25 个像素的大小。
之前的做法是,用的 yolov4 tiny 的网络,1920x1080resize 成 608x608 的输入,简单粗暴的标了几百个图片,两个 class,人和实心球。现在误检测很多,而且在实心球和背景颜色相近时也很难检测出来。现在后处理的逻辑是很多的,但是还是没法用,误检测太多,很多纯色的小区域,甚至足球排球都能识别成实心球。
我没想到什么办法,只能说现在多标一些数据,做做数据增强啥的。
有没有大佬提个建议,感谢!
1
Raven316 OP 大佬救我
|
2
leaves98 2021-04-29 23:27:09 +08:00 via Android
不是大佬,想到的:
1. 数据增强是必须的 2. 把原图切割成小块作为输入,当然小块边缘要有少量重叠 3. 试试 yolov5 |