增广数据集-图像旋转+标签坐标旋转

最新推荐文章于 2024-09-09 23:24:42 发布

weixin_44804536

最新推荐文章于 2024-09-09 23:24:42 发布

阅读量1.6k

点赞数

分类专栏： yolo 文章标签：标签旋转

本文链接：https://blog.csdn.net/weixin_44804536/article/details/89676801

版权

本文介绍了一种图像数据增强技术——图像旋转，并详细阐述了如何在旋转图像的同时，正确处理对应的标签坐标，以确保数据集的质量和模型训练的准确性。

摘要由CSDN通过智能技术生成

import numpy as np
import os
import cv2
from math import *


def rotate_images(path, angle):
    image_dir = path
    pathDir = os.listdir(image_dir)
    for s in pathDir:
        newDir = os.path.join(image_dir, s)
        # print(newDir)
        if os.path.isfile(newDir):
            if os.path.splitext(newDir)[1] == ".jpg":
                # lenth = len(newDir)
                # print(lenth)
                filename = int(os.path.basename(newDir)[:-4]) + 270

                """
                imgs = cv2.imread(newDir)
                imgs = skimage.transform.rotate(imgs, angle)
                # now = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
                plt.figure("rotated")
                plt.imshow(imgs)
                plt.axis("off")
                plt.savefig("D:/learn/lianxi/python_cnn/save/"+str((i+17)) + ".jpg")