#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Sep 25 09:43:57 2018
@author:
"""
from PIL import Image
from pylab import *
import os
import shutil
dirlis = '/home/zq/Desktop/train_rgb/'
dirsave = '/home/zq/Desktop/train_rgb_plus/'
for names in os.listdir(dirlis):
namesplit = names.split('.')
if(namesplit[1]=='png'):
#读取图片并转为数组
img1 = array(Image.open(dirlis+namesplit[0]+'.png'))
img2 = array(Image.open(dirlis+namesplit[0]+'.png'))
#BGR --> RGB
img1 = img1[:,:, (2, 1, 0)]
imsave(dirsave+namesplit[0]+'rb.png', img1)
#BGR -->BRG
img2 = img2[:,:, (1, 0, 2)]
imsave(dirsave+namesplit[0]+'rg.png', img2)
else:
shutil.copyfile(dirlis+names,dirsave+namesplit[0]+'rb.json')
shutil.copyfile(dirlis+names,dirsave+namesplit[0]+'rg.json')