yolov5半自动打标签(opencv版本),识别目标画框并将坐标信息保存在xml中

文章目录

    • 1.yolov5预训练模型推理
    • 2. opencv边缘检测
      • 结果展示

yolov5训练数据集时,需要对数据进行打标签,可以通过两种方法进行半自动化打标签。

1.yolov5预训练模型推理

yolov5预训练模型:将待打标签的图片输入预训练模型中进行推理,将推理结果保存在txt文件中,然后将txt文件转化为xml文件,最后导入labelimg中进行校准和校证。

具体txt转xml流程在本人之前的博客已经展示,链接如下;
https://blog.csdn.net/qq_42754919/article/details/129436134

上述方法需要有预训练模型,有一定的局限性,若首次训练模型,则无法使用。

针对上述问题,本博客提供了第二种方法。

2. opencv边缘检测

使用opencv进行边缘检测,然后找到外接矩形框,根据返回的坐标信息写入xml文件。

缺点:只能针对目标画一个完整的框

你可能感兴趣的:(深度学习,opencv,YOLO,计算机视觉,人工智能)