(python) 폴더 내 PNG를 Numpy array로 바꿔 저장하는 프로그램(첨부파일)

PNG to nparray.ipynb

0.00MB

영상 dataset 에서 PNG를 Numpy 라이브러리가 제공하는 array(python은 배열이라는 자료형이 존재하지 않습니당)로 만들어 저장할 경우가 생겼다.

 

참고로 필자는 Jupyter notebook에서 작업했다.

 

따라서 간단히 폴더 안에 있는 모든 PNG file들을 Numpy array로 바꾸어 다른 폴더에 저장해주는 프로그램을 짜보았다.

 

path_dir = '' 에서 ''의 안을 파일의 주소로 채운다. (파일의 주소를 찾고, 쓰는 법은 글 하단에 나와 있다.)

np.save("" + png, pixel) 에서 ""의 안을 저장할 파일의 주소로 채운다.

import numpy as np
from PIL import Image
import os


path_dir = ''# '' 안에 파일을 묶고 있는 "폴더 경로"를 쓰세요
file_list = os.listdir(path_dir)

for png in file_list:
    image = Image.open(path_dir + png)
    pixel = np.array(image)
    png = png.split('.')[0]
    np.save(""+png, pixel) #저장할 '폴더 경로'를 쓰세요

 

 

<파일 주소를 찾고 쓰는 법>

1. PNG가 들어있는 폴더로 들어간다.

2. PNG 하나를 마우스 우 클릭 한다.

3. 속성(R)을 클릭한다.

 

4. 3번까지 하면 아래와 같은 창이 뜰 것이고, "위치" 오른쪽에 나와있는 주소를 드래그하고 ctrl + C 한다. (주소 복사)

 

5. 저장한 주소를 python 코드의 path_dir = ' ' 의 ' '사이에 ctrl + V 하여 붙여 넣는다.

밑에 np.save 부분의 " "사이도 같은 맥락으로 처리해 주면 된다.