显著性检测数据集及其扩充

现有公开的数据集
转载自数据集
通过水平翻转扩充数据集。
水平翻转扩充数据集：

import os
import numpy as np
from PIL import Image

def flip(root_path,img_name):   #翻转图像
    img = Image.open(os.path.join(root_path, img_name))
    filp_img = img.transpose(Image.FLIP_LEFT_RIGHT)
    #filp_img.save(os.path.join(root_path,img_name.split('.')[0] + '_flip.jpg'))
    return filp_img


def createImage(imageDir,saveDir):
   i=0
   for name in os.listdir(imageDir):
      i=i+1
      saveName1 = "flip" + str(i) + ".jpg"
      saveImage1 = flip(imageDir,name)
      saveImage1.save(os.path.join(saveDir, saveName1))

if __name__ == '__main__':

    imageDir="G:/dataset/ceshi/1" #要改变的图片的路径文件夹
    saveDir="G:/dataset/ceshi/4"   #数据增强生成图片的路径文件夹
    createImage(imageDir,saveDir)