【lpliner】【albumentations】【lambda】【python】【旋转图片增加数据集`个数`】

写在前面

• 这次的code是因为在训练过程中数据量不足，在FCN训练时添加了transform，可惜结果不是我想要的。遂，更换了Unet，可惜在训练中显示n_train过小（len(train_loader)）, 于是就想着增加数据集。所以出现了以下代码
• 这里的transform主要就是旋转，就是旋转，还都是90°的倍数
• 话不多说上代码，不喜请大力喷。

---

Code

#!/usr/bin/python3
# coding: utf-8
# @Time: 2021/3/10 11:39
# @Project: SemanticSegmentation
# @File: create_more_dataset.py
# @Author: Lpliner
# @Description: 将原始图片经过90,180,270°旋转，以及经过水平翻转后再次进行90， 180， 270旋转，以此增大数据量 每张图片*8

import os
import cv2
import shutil
import logging
import albumentations as A
from matplotlib import pyplot as plt

logging.basicConfig(level=logging.INFO, format='%(levelname)s: %(message)s')

# settings
base_dir = "../dataset/train"           # 数据原始所在位置
save_dir = "./train"                    # 数据另存为的位置
if not os.path.exists(save_dir): os.mkdir(save_dir)
save_image_dir = os.path.join(save_dir, "image")        # 新数据二级目录
if not os.path.exists(