Image-rotate-brightness-blur

来源：互联网发布：美国人口老龄化数据编辑：程序博客网时间：2024/05/20 07:20
import osimport globfrom PIL import Imageimport cv2# files = glob.glob('./*/*.png')# for f in files:#     cv2.imwrite(os.path.splitext(f)[0] + '.jpg', cv2.imread(f))#     os.remove(f)# left-right top-bottom transposesets = './train'  # 'val'####################### rotation 8 directions ####################################files = glob.glob('./{0}/*/*.jpg'.format(sets))length = len(files)for i, f in enumerate(files):    print 'left-right top-bottom transpose of {0} in {1}, {2}'.format(i, length, f)    im = Image.open(f)    out = im.transpose(Image.FLIP_LEFT_RIGHT)    outName = os.path.splitext(f)[0] + 'LR.jpg'    out.save(outName)files = glob.glob('./{0}/*/*.jpg'.format(sets))length = len(files)for i, f in enumerate(files):    print '90, 180, 270 transpose of {0} in {1}, {2}'.format(i, length, f)    im = Image.open(f)    # rotation 90    out = im.transpose(Image.ROTATE_90)    outName = os.path.splitext(f)[0] + 'R90.jpg'    out.save(outName)    # rotation 180    out = im.transpose(Image.ROTATE_180)    outName = os.path.splitext(f)[0] + 'R180.jpg'    out.save(outName)    # rotation 270    out = im.transpose(Image.ROTATE_270)    outName = os.path.splitext(f)[0] + 'R270.jpg'    out.save(outName)###################### brightness 0.5 & 1.5 times ##############################files = glob.glob('./{0}/*/*.jpg'.format(sets))length = len(files)for i, f in enumerate(files):    print 'change to be darker and brighter, {0} in {1}, {2}'.format(i, length, f)    im = cv2.imread(f)        # darker 0.5 time    out = im * 0.5    outName = os.path.splitext(f)[0] + 'darker.jpg'    cv2.imwrite(outName, out)    # brighter 270    out = im * 1.5    outName = os.path.splitext(f)[0] + 'brighter.jpg'    cv2.imwrite(outName, out)####################### Gaussian blur  ##############################files = glob.glob('./{0}/*/*.jpg'.format(sets))length = len(files)for i, f in enumerate(files):    print 'Gaussian blur, {0} in {1}, {2}'.format(i, length, f)    im = cv2.imread(f)        out = cv2.GaussianBlur(im, ksize=(7, 7), sigmaX=10)    outName = os.path.splitext(f)[0] + 'blur.jpg'    cv2.imwrite(outName, out)print 'completed!'
阅读全文
0 0