【问题标题】:How to draw bounding boxes for predicted classes for a custom trained image classifier using fastai如何使用 fastai 为自定义训练图像分类器绘制预测类的边界框
【发布时间】:2019-05-10 07:51:56
【问题描述】:

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

【问题讨论】:

    标签: bounding-box fast-ai


    【解决方案1】:

    所以我会分两步完成。首先,学习如何在 fastai 中获取边界框。示例here

    如果你有,你可以使用 opencv 逐帧加载视频并标记它。 OpenCV example.

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

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2019-09-28
      • 2020-02-02
      • 2011-04-10
      • 2019-06-19
      • 1970-01-01
      • 2019-10-15
      相关资源
      最近更新 更多