어떤 디렉토리 내에 존재하는 모든 이미지 파일들을 cv2.imread 등의 방식으로 읽고 싶다면, 일단 각 이미지 파일의 경로를 알아야합니다. 저는 그것을 리스트에 하나씩 담아주도록 하겠습니다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
import os
root_dir = 'C:/Users/Sim/Desktop/test' # 디렉토리
img_path_list = []
possible_img_extension = ['.jpg', '.jpeg', '.JPG', '.bmp', '.png'] # 이미지 확장자들
for (root, dirs, files) in os.walk(root_dir):
if len(files) > 0:
for file_name in files:
if os.path.splitext(file_name)[1] in possible_img_extension:
img_path = root + '/' + file_name
# 경로에서 \를 모두 /로 바꿔줘야함
img_path = img_path.replace('\\', '/') # \는 \\로 나타내야함
img_path_list.append(img_path)
print(img_path_list)
|
cs |
위와 같이 코딩해주면, 바탕화면에 있는 test 디렉토리(C:\Users\Sim\Desktop\test) 내에 존재하는 모든 이미지 파일들의 경로를 리스트의 형태로 얻을 수 있습니다. 참고로 test 디렉토리 내부는 다음과 같은 트리 구조를 갖고 있습니다.
위 코드를 통해 얻은 test 디렉토리 내 이미지들의 경로 리스트입니다.
만약 디렉토리 내에 다른 확장자의 이미지들이 존재한다면, possible_img_extension 리스트에 추가해주시면 됩니다.
'코딩 > python + 컴퓨터비전' 카테고리의 다른 글
[ubuntu+python] 얼굴 인식하기 (face_recognition 라이브러리 설치부터 사용까지) (2) | 2021.01.19 |
---|---|
[python] cv2.imread, cv2.imwrite 한글 경로 인식을 못하는 문제 해결 방법 (0) | 2021.01.08 |
[python] 어떤 디렉토리 내 존재하는 모든 이미지 파일들의 경로 리스트 만들기 (0) | 2021.01.08 |
[python] 쉽고 간단하게 마스크 착용 유무 판별기 만들기 (0) | 2021.01.07 |
[python] 웹캠 영상에서 검출된 얼굴들에 모자이크 효과 넣기 (0) | 2020.09.25 |
[python] 채워진 사각형 그리는 방법, cv2.rectangle 함수 활용 (0) | 2020.09.10 |
[python] 단방향 또는 양방향으로 화살표 있는 선 그리기, cv2.arrowedLine 함수 활용 (0) | 2020.09.09 |
[python] 사진에 카툰(cartoon) 효과주기, cv2.stylization 함수 (0) | 2020.09.09 |
[python] 이미지 회전시키기, cv2.rotate 함수 (0) | 2020.09.07 |