학습 후 색이 달라지는 문제

카메라 이미지 품질 향상 AI 경진대회

학습 후 색이 달라지는 문제

2021.07.22 15:48 2,599 Views

안녕하세요.

모델 학습 후 loss는 낮으나 색이 다르게 나오는 경우가 있습니다.

손실함수는 mse를 사용하였는데, 이경우 어떤 부분들을 살펴보면 도움이 될까요?

로그인이 필요합니다

comment

0 / 1000

INVI

2021.07.22 16:03

이미지 저장하실때, BGR로 저장되신거 같아요.
cv2.cvtColor(img, cv2.COLOR_BGR2RGB) 모듈 사용해보시는걸 추천합니다.

뇨로롱

2021.07.22 16:12

답변 감사합니다.
추론코드에 있는 것으로 실행한 것이라 cv2.cvtColor(img, cv2.COLOR_BGR2RGB)을 사용한 결과가 위 사진입니다.

TEAM-EDA

2021.07.22 16:31

그 이게 체크하셔야하는게 로드할때 BGR2RGB 제대로 되어있는지, 슬라이딩 윈도우로 자르고 저장할때 RGB로 하셨는지 다 체크해봐야해요 ㅠㅠ 저도 초반에 이 부분 헷갈려서 학습 잘 안되는 문제가 있었습니다..

뇨로롱

2021.07.22 17:39

그 부분이 잘못되어있으면 항상 제대로 나와야 하는데, 어쩔 때는 색이 잘 나오고 어쩔 때는 잘 안 나오는 것이 문제네요.

파이썬초보만

2021.07.23 12:45

저같은 경우는 모델 체크포인트를 안가져와서 저런 현상이 나타나더라구요

현재 글

학습 후 색이 달라지는 문제

Competition - 카메라 이미지 품질 향상 AI 경진대회

Likes 5

Comments 5

3년 전