파일이름 리스트 텍스트 파일로 저장하기

참고하는 코드를 사용해서 dataset을 만들고 나니 모든 이미지들이 뒤죽박죽 되어서..

txt로 제목 정리를 해서 불어와야겠다

root = "/home/cityscapes" 

# 파일명 리스트로 저장
rgb_paths = glob(os.path.join(root, 'leftImg8bit', 'train', '**/*.png'))

# aachen_000000_000019_disparity -> aachen_000000_000019 로 변환
def get_name(file_path):
    base_name = os.path.basename(file_path)
    prefix = base_name.split('_')[0] + '_' + base_name.split('_')[1] + '_' + base_name.split('_')[2]
    return prefix

# 적용
prefixes = [get_name(path) for path in rgb_paths]

# txt파일에 작성
with open('/home/cityscapes/train.txt', 'w') as file:
    for item in rgb_prefixes:
        file.write(f"{item}\n")