我使用fastai模型resnet34创建了枪炮图像分类器。我能够预测图像中的课程,但我想扩展它以对视频素材进行实时分析。我想在枪支刀周围装上边界盒,但无法理解如何得到它。有人可以引导我通过。
答案 0 :(得分:1)
所以我分两步来做。首先,了解如何在Fastai中获取边界框。示例here。
如果有,则可以使用opencv逐帧加载视频并对其进行标签。 OpenCV example。
同时拥有两者时,就很容易实时进行操作。只需标记从网络摄像头加载的帧即可。此过程非常昂贵,因此您可能需要在推断之前降低图像分辨率。我已经在tensorflow中完成了它,并且效果很好。 This repository was extremely useful