[Python] Dissolve Folder(폴더 해체)

dada·2024년 10월 23일

Comprehensive Project

목록 보기
2/2
post-thumbnail

데이터셋은 디지털 카메라, 스마트폰, 스마트 패드로 촬영한 것들이 다 폴더로 구분되어 있으며 사람 한 명 한 명 하위폴더로 나뉘어져 있기 때문에 이 모든 사진/json 파일을 복사해서 새로운 폴더에 저장하기 위해서 만든 코드

  • 이 코드는 image labelling 작업 시 쓰이며, json 파일을 통합하는 경우 또한 사용된다.

## 폴더 안에 있는 파일들을 복사해서 새로운 폴더에 저장

import os
import shutil

copy_file_folder = os.path.join(os.getcwd(), "All_json_file") # 해체해서 저장할 폴더 
print(copy_file_folder)

def is_folder(file_path):
    path = file_path
    for i in os.listdir(path):
        if os.path.isdir(path+"\\"+i):
            print(path+"\\"+i)
            is_folder(path+'\\'+i)
        elif(path+"\\"+i).endswith('.json'):
            try:
                shutil.copyfile(path+"\\"+i, copy_file_folder+"\\"+i)
                print(path+"\\"+i)
            except:
                pass


# 해체할 폴더 대상
folder = ["phone", "digit", "pad"]

for i in folder:
    looking_path = i

    for i in os.listdir(looking_path):
        if os.path.isdir(looking_path+'\\'+i):
            is_folder(looking_path+"\\"+i)
            print(f"Finish folder : {i}")

profile
CV, Vision AI 등을 공부합니다

0개의 댓글