仅预测边界框答案

【问题标题】：Predict bounding box only仅预测边界框
【发布时间】：2019-06-25 05:24:49
【问题描述】：

我有两个文件夹训练和测试。每个文件夹有 14k 张随机单个对象的图像，如椅子、盒子、风扇、罐子等。除此之外，对于包含随机对象（边界框）的训练集的每个图像，我都有 4 列 [x1,x2,y1,y2]。有了这些信息，我想预测测试集的边界框。
我是计算机视觉方面的新手，如果有人能帮助我如何开始训练这种模型，那将非常有帮助。我找到了 yolov3，但它也包含分类。

【问题讨论】：

标签： deep-learning computer-vision object-detection

【解决方案1】：

我推荐你看github代码here。

在detect.py中，有do_detect()函数。

此函数返回您想从图像中获取的类和 bbox。

boxes = do_detect(model, image, confidence_threshold, nms_threshold)

【讨论】：