이미지 파일 하나씩 조회하기

2020. 6. 30. 19:38파이썬

import cv2
import os

index = 0
file_list = os.listdir('.')
file_list_jpg = [file for file in file_list if file.endswith(".jpg")]
key = 0
chkey = ''

while chkey != 'Q' and chkey != 'q':
    img = cv2.imread(file_list_jpg[index], cv2.IMREAD_COLOR)
    cv2.imshow('image', img)
    key = cv2.waitKey(0)
    chkey = chr(key)
    #print('key : ', key, chr(key))
    if chkey == 'n':
        index = index + 1
        if index >= len(file_list_jpg):
            index = 0
    elif chkey == 'p':
        index = index - 1
        if index < 0:
            index = len(file_list_jpg) - 1
    print(index, len(file_list_jpg))        


cv2.destroyAllWindows() 

매번 새로 만드는게 귀찮아서 ...