1

我使用 fastai 模型 resnet34 创建了枪支和刀具图像分类器。我能够预测图像中的类别，但我想扩展它以实时分析视频片段。我想在枪支和刀具周围设置边界框，但无法理解如何获得它。有人可以指导我完成。

4

1 回答 1

1

所以我会分两步做。首先，学习如何在 fastai 中获取边界框。这里的例子。

如果你有，你可以使用 opencv 逐帧加载视频并标记它。OpenCV 示例。

当你同时拥有两者时，它很容易实时完成。只需标记从您的网络摄像头加载的帧。这个过程非常昂贵，因此您可能希望在推理之前降低图像分辨率。我已经在 tensorflow 中完成了它，并且效果很好。这个存储库非常有用

于 2019-05-14T18:14:04.323 回答