AD

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

노른자랑흰자
2020-03-26 13:46:41 274 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개  
이전 댓글 더 보기
공지잡담최신뉴스자료공유연구/플젝Q&A
1
잡담
여기 죽었나요? [2]
노른자랑흰자
11-12
7
잡담
딥게더 죽지 않았다면 정답을 알려줘 [1]
공포게임초고수서나랑
10-29
1
잡담
질문하나 드립니다 [2]
루버거
09-25
13
08-22
1
잡담
캐글하는 트수들 있나요? [2]
이탈예측생존분석
06-02
3
05-21
3
잡담
Papers with Code [1]
참신한기획
04-29
11
04-22
8
자료공유
4월 16일 논문 리뷰 방송 [1]
Moderator 베일쿤
04-15
4
04-10
3
최신뉴스
Ian Goodfellow 애플로 이직 [2]
Global Moderator gamingryan235
04-05
7
04-05
인기글 글 쓰기