python Corrupt JPEG data: premature end of data segment warning 이미지 깨짐 확인하기

Corrupt JPEG data: premature end of data segment warning은 warning이기 때문에 학습할 때 문제가 되지 않음.

 

하지만 위 warning은 이미지가 깨진것을 의미함. 

이를 위해 cv2.imread()를 사용하여 모든 이미지를 확인하면서 try: ~ exception:~ 로 어떤 이미지 경로인지 찾아낼 수 없음. (나는 못했음) 왜냐하면 에러가 아니니까 try에서 못잡음. 

 

결론적으로 내가 해결한 방법은 아래와 같다.

from skimage import io 를 임포트하고 아래와 같이 이미지를 불러오면 깨진 이미지에서 에러가 잡힘.

이때 except 에서 에러가 뜬 이미지 경로가 뜨게 하면됨.

from skimage import io
try:
	img = io.imread(img_file)
except:
	print("Cannot load : {}".format(img_file))