AD

잡담 (트수형들 도와줘)Image segmentation 감 좀 잡게 알려주세요..

노른자랑흰자
2020-03-26 13:46:41 1736 4 13

다 죽은 트게더지만 가아끔 눈팅하는 사람들 있는거같아서 써봅니다 ㅎㅎ..(트수형들 도와줘)

데이터 셋을 이미지랑 마스크 데이터 이렇게 두개로 하고

이미지를 모델(u-net)에 넣은 다음 마스크 데이터랑 바이너리 크로스 엔트로피 로스를 써서 트레이닝을 했는데요

지금 로스가 0.0067정도로 나올때까지 학습을 시켰어요.

그래서 다음으로 얼마나 잘됬나 확인해보고싶어서

predicted mask를 이미지로 확인하고싶은데 잘 안되고있어요..


pred_mask = model(img)에서 

pred_mask 를 적절히 차원 변경해서 matplotlib으로 출력했는데 이미지가 너무 엉뚱하게 난해한 그림으로만 나와서요..

그래서 제가 개념을 잘못이해하고있는가해서 그런데

예상된 이미지 == 이미지를 모델에 넣고 나온 것 이 아닌가요?


수정)

방금 생각난건데 흑백으로 0이랑 1로 나오게 하는데 

0.5를 threshold로 해서 그 밑으로는 검정색 위로는 흰색으로 나타내게 따로 짜야되는거에요?

밑에 나온 그림은 차례대로 image, mask, predicted mask입니다.


c14bac3dfc13d938851188ad06afcb71.png

후원댓글 13
댓글 13개  
이전 댓글 더 보기
TWIP 잔액: 확인중
공지잡담최신뉴스자료공유연구/플젝Q&A
0
04-29
0
10-04
0
잡담
안녕하세요
김재진
08-31
0
잡담
요즘 딥게더...
제리쿤
10-30
0
잡담
딥러닝은 뭔가요?
신혈득순
09-12
1
잡담
여기 죽었나요? [2]
노른자랑흰자
11-12
1
잡담
질문하나 드립니다 [2]
히나신
09-25
13
08-22
1
잡담
캐글하는 트수들 있나요? [2]
이탈예측생존분석
06-02
3
05-21
3
잡담
Papers with Code [1]
참신한기획
04-29
인기글 글 쓰기