我正在计算视频中车辆的总编号,但我想只检测正在向上行驶的车辆(道路有分隔线(,所以我的观点是,我可以只在车辆向上移动的矩形上使用yolo吗?我不想检测道路另一边的车辆。
有没有办法像我可以画一个矩形并只检测该特定矩形上的对象?
我能想到的最好的是对于每一帧,我必须裁剪帧,执行所有操作并将其拼接回原始帧。我期待一个更简单的替代方案
任何帮助,不胜感激。谢谢
您可以像往常一样对整个图像执行 yolo,但添加一个 if 条件以仅绘制中心位于特定区域的框。或者,您可以将此条件(位置(添加到 IoU 的条件旁边(过滤检测到的框(。此外,您还可以根据移动车辆的方向进行单独计数,并为两个方向使用两个不同的计数器。
如果你不介意我问,你是如何跟踪车辆的?
我正在做类似的事情...
如果您的产品要像灯光轮询一样固定,那么很明显,您可以通过训练模型来检测道路和斑马线。
或
手动输入这些值...
稍后仅在帧的这些部分上运行对象检测和对象跟踪,即使用
frame[ymax:ymin, xmax:xmin]
这会减小图像大小,从而提高处理速度。
但是,为什么在工作后再次需要完整图像呢? 但是,如果您确实需要它,那么您只需将道路上物体检测框的XMIN和Ymin的值添加到该对象检测框中检测到的车辆的边界框中,即可在未裁剪的图像中获取其边界框值。