본문 바로가기

카테고리 없음

파이썬 폴더 내 이미지 파일 이름 불러오기 베이스 코드

import os
from glob import glob

filename_list = glob(os.path.join("./test_real", "*.jpg"))

def getonlyfilename(path):
  filename = path.split('/')[-1]
  return filename

def rm_extension(filename):
  filename_rm_ext = filename[:-4]
  return filename_rm_ext


for filepath in filename_list:
  #print(filepath)
  filename = getonlyfilename(filepath)
  #print(filename)
  filename_rm_ext = rm_extension(filename) # for only image type  or .txt
  print(filename_rm_ext)