在每次迭代中选择相同顺序的图像

问题描述

我一直在尝试使用下面的代码来扩充图像数据以进行语义分割，但是输出看起来很相似。选择并处理了相同的图像，并且输出文件夹中大部分包含重复的图像。我不确定这个问题来自哪里

from PIL import Image
from skimage import io,img_as_ubyte
from scipy.ndimage import rotate
import random
import cv2
from scipy import *
import skimage as sk

images_path= '/images/'
mask_path = '/masks/'
Augmentate_images= '/Aug_images/'
Augmentate_masks= '/Aug_masks/' 
seed_for_random = 44

#Defining custom functions for each operation


def rotation(image,seed):
    random.seed(seed)
    angle= random.randint(-180,180)
    r_img = rotate(image,angle,mode='reflect',reshape=False,order=0)
    return r_img

def h_flip(image,seed):
    hflipped_img= np.fliplr(image)
    return  hflipped_img

def v_flip(image,seed):
    vflipped_img= np.flipud(image)
    return vflipped_img

def v_transl(image,seed):
    random.seed(seed)
    n_pixels = random.randint(-64,64)
    vtranslated_img = np.roll(image,n_pixels,axis=0)
    return vtranslated_img

def h_transl(image,64)
    htranslated_img = np.roll(image,axis=1)
    return htranslated_img

transformations = {'rotate': rotation,'horizontal flip': h_flip,'vertical flip': v_flip,'vertical shift': v_transl,'horizontal shift': h_transl}

#================================================
IMAGES=[] # to store paths of images from folder
MASKS=[] # to store paths of masks from folder
for im in os.listdir(images_path):  
    IMAGES.append(os.path.join(images_path,im))
for msk in os.listdir(masks_path):  
    MASKS.append(os.path.join(masks_path,msk))


#=============================
num_files_desired = 1000
num_generated_files = 1 # variable to iterate till images_to_generate

while num_generated_files <= num_files_desired:
    # random image from the folder
    number = random.randint(0,len(IMAGES)) 
    IM = IMAGES[number]
    MSK= MASKS[number]
    original_image = io.imread(IM)
    original_mask = io.imread(MSK)
    num_transformations = None
    transformed_image = None
    #================
    n = 0       #variable to iterate till number of transformation to apply
    transformation_count = random.randint(1,len(transformations)) 
    while n <= transformation_count:
        key = random.choice(list(transformations)) #randomly choosing method to call
        seed = random.randint(1,100)  #Generate seed to supply transformation functions. 
        transformed_image = transformations[key](original_image,seed)
        transformed_mask = transformations[key](original_mask,seed)
        n = n + 1
    new_image_path= "%s/augmented_image_%s.jpg" %(img_augmented_path,num_generated_files)
    new_mask_path = "%s/augmented_mask_%s.png" %(mask_augmented_path,num_generated_files)  
    io.imsave(new_image_path,transformed_image)
    io.imsave(new_mask_path,transformed_mask)
    num_generated_files += 1`

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）