# -- coding: UTF-8 --
'''
Author: Clouds rising
Date: December, 2023
QQ: 504156006
'''

import os
import glob
import numpy as np
from PIL import Image

# 要修改的文件夾路徑
input_dir = 'label_png'

# 新文件夾的路徑
output_dir = 'label_npy'

# 創(chuàng)建輸出文件夾
if not os.path.exists(output_dir):
    os.makedirs(output_dir)

# 遍歷文件夾中的所有png圖像
for img_path in glob.glob(os.path.join(input_dir, '*.png')):

    # 打開原始的png圖像
    img = Image.open(img_path)

    # 轉(zhuǎn)換圖像
    img_8bit = img.convert('L') #參數(shù)L: 8位像素，黑白

    # 創(chuàng)建一個(gè)轉(zhuǎn)換表，將灰度值為255的像素點(diǎn)變?yōu)?
    '''
    在這一行代碼中，創(chuàng)建了一個(gè)包含256個(gè)元素的列表（0到255的灰度值）。該列表的每個(gè)元素都通過條件判斷
    0 if i == 255 else i 進(jìn)行賦值。這意味著如果灰度值為255，那么賦值為0，否則保持不變。
    這樣就形成了一個(gè)轉(zhuǎn)換表，用于將灰度值為255的像素點(diǎn)變?yōu)?。
    '''
    invert_table = [0 if i == 255 else i for i in range(256)]
    '''
    在這一行代碼中，通過point方法應(yīng)用了之前創(chuàng)建的轉(zhuǎn)換表。
    這個(gè)方法會(huì)對(duì)圖像中的每個(gè)像素點(diǎn)進(jìn)行變換，根據(jù)轉(zhuǎn)換表將灰度值為255的像素點(diǎn)變?yōu)?，而其他灰度值保持不變。
    最終，得到了經(jīng)過反轉(zhuǎn)處理的圖像img_inverted1。
    '''
    img_inverted1 = img_8bit.point(invert_table, 'L')

    # 將圖像對(duì)象轉(zhuǎn)換為NumPy數(shù)組
    img_array = np.asarray(img_inverted1)

    # 生成與輸入文件相同的輸出文件名
    output_filename = os.path.join(output_dir, os.path.basename(img_path).replace('.png', ''))

    # 自動(dòng)保存為npy文件
    np.save(output_filename, img_array)

from PIL import Image
import numpy as np

#讀取PNG圖像
img = Image.open('image.png')

# 轉(zhuǎn)換為RGB格式
img = img.convert('RGB')

# 轉(zhuǎn)換numpy數(shù)組
img_arr = np.array(img)

# 保存為.npy格式
np.save('image.npy', img_arr)