首页 > 解决方案 > YOLO V4 Tiny - 从一张带注释的图像中制作更多照片

问题描述

我正在尝试使用 google collab 制作一个 yolo v4 微型自定义数据集。我正在使用 labelImg.py 进行图像注释,如https://github.com/tzutalin/labelImg所示。

我已经注释了一张图片,如下所示,

注释图像

带有注释坐标的 .txt 文件如下所示,

0 0.580859 0.502083 0.303906 0.404167

我只有一门课,就是计算器课。我想用这一张图片再生成 4 张带注释的图片。我想每次将带注释的图像旋转 45 度并创建一个新的带注释的图像和 a.txt 坐标文件。我在 roboflow 中看到过类似的事情,但我不知道如何使用 python 脚本手动完成。有可能做到吗?如果有怎么办?

标签: pythonimage-processingcomputer-visionobject-detectionyolo

解决方案


您可以查看下面的 repo 和文章,了解基于 python 的数据增强,包括旋转、剪切、调整大小、平移、翻转等。

https://github.com/Paperspace/DataAugmentationForObjectDetection

https://blog.paperspace.com/data-augmentation-for-bounding-boxes/

如果您正在使用AlexeyAB 的 yolov4 暗网存储库,那么您可以使用一些增强功能来增加训练数据的大小和变化。

https://github.com/AlexeyAB/darknet/wiki/CFG-Parameters-in-the-%5Bnet%5D-section

查看Data augmentation部分,您可以通过将它们添加到 yolocfg文件来使用各种定义的增强进行对象检测。


推荐阅读