图片颜色通道change

最新推荐文章于 2025-03-02 16:43:10 发布

原创最新推荐文章于 2025-03-02 16:43:10 发布 · 836 阅读

CC 4.0 BY-SA版权

#!/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')