【发布时间】:2019-05-10 07:51:56
【问题描述】:
我使用 fastai 模型 resnet34 创建了枪支和刀具图像分类器。我能够预测图像中的类别,但我想扩展它以实时分析视频片段。我想在枪支和刀具周围设置边界框,但无法理解如何获得它。有人可以指导我吗?
【问题讨论】:
标签: bounding-box fast-ai
我使用 fastai 模型 resnet34 创建了枪支和刀具图像分类器。我能够预测图像中的类别,但我想扩展它以实时分析视频片段。我想在枪支和刀具周围设置边界框,但无法理解如何获得它。有人可以指导我吗?
【问题讨论】:
标签: bounding-box fast-ai
所以我会分两步完成。首先,学习如何在 fastai 中获取边界框。示例here。
如果你有,你可以使用 opencv 逐帧加载视频并标记它。 OpenCV example.
当你同时拥有这两者时,实时操作就很容易了。只需标记从您的网络摄像头加载的帧。此过程非常昂贵,因此您可能希望在推理之前降低图像分辨率。我已经在 tensorflow 中完成了它,并且效果很好。 This repository was extremely useful
【讨论】: