推荐一个无比强大的数据增强库，mask图像同时增强

刘泽豪真讨厌

于 2019-11-27 14:05:59 发布

阅读量1.2k

点赞数

分类专栏： AR深度学习项目

本文链接：https://blog.csdn.net/liu506039293/article/details/103274698

版权

博客推荐了imgaug库，这是一个强大的图像数据增强工具，适用于图像处理和机器学习项目。作者分享了一段使用该库的代码示例。

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

送一份我用的代码

'''
@Author: haoMax
@Github: https://github.com/liuzehao
@Blog: https://blog.csdn.net/liu506039293
@Date: 2019-11-27 11:44:45
@LastEditTime: 2019-11-27 14:01:14
@LastEditors: haoMax
@Description: 
'''
import numpy as np
import imgaug.augmenters as iaa
import os
import cv2
PIXEL=512
class_num=4
X_CHANNEL=3
Y_CHANNEL=1
def get_all_files(bg_path):
    files = []

    for f in os.listdir(bg_path):
        if os.path.isfile(os.path.join(bg_path, f)):
            files.append(os.path.join(bg_path, f))
        else:
            files.extend(get_all_files(os.path.join(bg_path, f)))
    files.so