Scroll indicator done
728x90

딥러닝 모델을 돌리고 인풋 적용 후 결과 이미지를 저장하는 데 계속 까만 사진으로 저장이 되었다 ..

-> 정규화가 안되있던 것 (정규화 : 특정 부분에 몰려 있는 값을 전체 영역으로 골고루 분포하게 하도록 만듦)

imgs = np.expand_dims(load_net_in(INPUT_IMG_FNAME, desired_size=INPUT_SIZE), axis=0)
ys = model.predict(imgs)
y = postprocess(ys)[0]

save_file = "x" + str(j) + ".jpg"


y = cv2.cvtColor(y, cv2.COLOR_BGR2RGB)
y = cv2.normalize(y, None, 0, 255, cv2.NORM_MINMAX, dtype=cv2.CV_8U) # 이 부분
cv2.imwrite(save_file, y)

 

728x90