Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | 6 | 7 |
8 | 9 | 10 | 11 | 12 | 13 | 14 |
15 | 16 | 17 | 18 | 19 | 20 | 21 |
22 | 23 | 24 | 25 | 26 | 27 | 28 |
29 | 30 |
Tags
- 도커
- python최단거리
- Python
- graph
- pyvista
- 지하철역좌표
- 동명이인찾기
- geopandas
- 3d
- GNN
- 그리드분할
- 귀여운고래
- 파이썬
- GCN
- 이미지빌드
- 폴더조사
- STL
- MESH
- 도커 레이어
- 3d데이터
- geojson
- 데이터입수
- GIS
- osmnx
- 패치분할
- docker
- 알고리즘
- Set
- 좌표거리
- 컨테이너
Archives
- Today
- Total
이것저것 기록
[python] 폴더에서 원하는 파일 형태만 불러오기 본문
특정 폴더에서 내가 원하는 파일 형태만 불러오기!
나같은 경우 특정 이미지 파일 형태만 불러와야할 때가 있다.
예를 들어 .jpg, .png 등의 이미지 파일 형태가 섞인 폴더에서 .png 파일 형태만 불러오고 싶을 때!
import os
import cv2
우선 필요한 라이브러리를 import 해오자.
path = 'C:/Users/user/Desktop/folder/' # 폴더 경로
os.chdir(path) # 해당 폴더로 이동
files = os.listdir(path) # 해당 폴더에 있는 파일 이름을 리스트 형태로 받음
files 라는 리스트에는 다음과 같은 파일 이름이 담겨있다.
리스트에 담긴 것은 어디까지나 '파일명'이기 때문에, 이미지를 이미지 파일로 불러오기 위해선 추가 작업이 필요하다.
png_img = []
jpg_img = []
for file in files:
if '.png' in file:
f = cv2.imread(file)
png_img.append(f)
if '.jpg' in file:
f = cv2.imread(file)
jpg_img.append(f)
cv2.imread('파일 이름')을 사용하여 이미지 파일을 불러올 수 있다.
이렇게 실행해주면 png_img에는 .png 이미지 파일이, jpg_img에는 .jpg 이미지 파일이 numpy array 형태로 들어가게 된다.
'코린이 > 코딩 기초 & 알고리즘 공부' 카테고리의 다른 글
[python] 딕셔너리(dict) 기초 (0) | 2020.10.09 |
---|---|
[python] print 문자열 formatting (0) | 2020.10.09 |
[python] list to array, array to list (0) | 2020.10.08 |
[python] 두 개 이상의 list 비교하기 (0) | 2020.10.08 |
[python] 디렉토리 경로 - 절대경로, 상대경로, 현재경로 (0) | 2020.09.22 |
Comments