[파이썬] 여러 파일 이름 한번에 변경하는 프로그램 만들기(feat. os.rename)

계기

YOLO학습을 위해 이미지 라벨링을 하니, 이미지 수백 장의 이름을 변경해야 되는 일이 생겼다.

하나하나 변경하면 되지만 너무 귀찮고 오래 걸린다.

파이썬으로 만들 수 있을 거 같아, 간단하게 코드를 짜봤다.

프로그램이라 하기엔 코드가 너무 짧긴 하지만, 뭐 내가 필요에 의해 만든 거니 유의미하지 않을까?라고 생각한다.

 

코드

import os

#기존 사진 백업 필수!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
#기존에 있는 사진들을 삭제됨

#기존 파일 경로
folder_path = 'C:\\Users'

#새로 저장 경로 (정확히 맞아야 함)
dst_path = 'C:\\Users'

#폴더안에있는 파일들 이름 리스트로 가져오기
file_names = os.listdir(folder_path)

#확인
print(file_names)

#이미지 넘버링
numbering = 1

for file in file_names:
    f_name, f_format = file.split(sep = '.') #파일 이름, 파일 확장자 split

    if f_format == 'jpg': #사진이면
        if f_name + '.txt' in file_names: #라벨링이 된 사진이면
            #print(f_name, f_format)
            src_jpg = os.path.join(folder_path, f_name + '.jpg')
            src_txt = os.path.join(folder_path, f_name + '.txt')

            rename_jpg = '3_' + '1_' + str(numbering) + '.jpg' # (팀원순서)_(영상순서)_(이미지 순서) + .jpg
            rename_txt = '3_' + '1_' + str(numbering) + '.txt'

            numbering += 1 #넘버링 다음
            print(rename_jpg)

            dst_jpg = os.path.join(dst_path, rename_jpg)
            dst_txt = os.path.join(dst_path, rename_txt)

            os.rename(src_jpg, dst_jpg)
            os.rename(src_txt, dst_txt)

설명

일단 처음부터 얘기해 보면,

YOLO로 학습하기 위해서 이미지를 영상에서 프레임단위로 추출했다.

Streamlit

추출하면 1.jpg, 2.jpg, 3.jpg... 이런 식으로 생성된다.

 

그리고 labelImg를 사용해서 이미지 라벨링을 진행했다. 라벨링을 하면 1.txt, 2.txt 이런 형식으로 라벨링 정보가 txt파일에 저장된다.

txt파일에는 라벨링 클래스, 박스의 4 꼭짓점 좌표가 저장된다.

이제 jpg와 txt를 내가 원하는 형식으로 바꾸고 싶어 파이썬으로 코드를 작성했다.

 

#기존 파일 경로
folder_path = 'C:\\Users'

#새로 저장 경로 (정확히 맞아야 함)
dst_path = 'C:\\Users'

#폴더안에있는 파일들 이름 리스트로 가져오기
file_names = os.listdir(folder_path)

먼저 코드를 실행하면 기존에 있는 파일들이 사라지니까 백업 필수. 

 

바꾸고 싶은 사진들이 있는 폴더경로를 지정한다.

그리고 바꿔서 저장할 경로도 지정해 준다. 제대로 안 하면 저장경로가 꼬인다. 

 

os.listdir을 통해 폴더에 있는 파일들의 이름을 리스트로 가져온다.

[1.jpg, 1.txt, 10.jpg, 10.txt, 100.jpg,...]

이런 식으로 숫자 순서가 안 맞긴 하는데 나는 이미지 순서는 별로 중요하지 않아 그냥 진행했다.

순서를 맞추려면 코드를 추가로 작성해서하면 될 거 같다.

 

#이미지 넘버링
numbering = 1

for file in file_names:
    f_name, f_format = file.split(sep = '.') #파일 이름, 파일 확장자 split

    if f_format == 'jpg': #사진이면
        if f_name + '.txt' in file_names: #라벨링이 된 사진이면
            #print(f_name, f_format)
            src_jpg = os.path.join(folder_path, f_name + '.jpg')
            src_txt = os.path.join(folder_path, f_name + '.txt')

            rename_jpg = '3_' + '1_' + str(numbering) + '.jpg' # (팀원순서)_(영상순서)_(이미지 순서) + .jpg
            rename_txt = '3_' + '1_' + str(numbering) + '.txt'

            numbering += 1 #넘버링 다음
            print(rename_jpg)

            dst_jpg = os.path.join(dst_path, rename_jpg)
            dst_txt = os.path.join(dst_path, rename_txt)

            os.rename(src_jpg, dst_jpg)
            os.rename(src_txt, dst_txt)

for으로 폴더에 있는 각 파일이름을 가져온다. 파일은 xxx.xxx형식이니까 '.'을 기준으로 split 해준다.

 

폴더에는 라벨링 한 사진이 있고, 라벨링 안 한 사진이 있다.

라벨링을 하면 사진 파일 이름과 같은 .txt파일이 생성되니까, 이를 이용해서 라벨링 한 사진들을 분류해 준다.

 

확인 됐으면, 기존 파일들의 경로(src_)를 os.path.join으로 가져온다.

 

그리고 내가 바꾸고 싶은 이름(rename_)으로 만들고, 이걸 os.path.join으로 새로운 저장 경로(dst)를 만들어준다.

 

이제 os.rename으로 이름 변경해 주면 끝.

 

결과

 

후기

아이디어도 간단하고 구현도 쉽다. 그래도 나름 자동화니까 유의미하지 않을까 생각한다.

멀티버스의 나는 파일 하나하나 클릭하며 파일명을 바꾸고 있을까?