agyeong / Image_segmentation

Semantic Image Segmentation
1 stars 0 forks source link

semantic segmentation - pytorch #4

Closed agyeong closed 3 years ago

agyeong commented 3 years ago

semantic_segmentation

위 링크 참고해서 모델 돌려보는 중입니다.

저희가 쓰는 label data는 픽셀값이 들어있는데, 위 링크에는 이미지의 위치에 class 값이 들어있습니다. class 값으로 이루어진 이미지를 따로 만들어야할 것 같습니다.

다른 모델들도 픽셀 값이 아닌 class 값이 들어간 이미지가 필요할 것 같습니다...

아래 사진은 이미지를 출력한 결과인데, 왼쪽은 저희가 사용하는 픽셀 값이 들어있는 데이터이고 오른쪽은 클래스가 들어있는 위 링크에서 사용한 데이터입니다.

image

agyeong commented 3 years ago

DL3_v1 브랜치에 semantic-segmentation-tutorial-pytorch-ag/minicity/data_refactoring.ipynb 파일이 픽셀값을 클래스로 변환하는 파일입니다!

agyeong commented 3 years ago

픽셀값을 클래스값으로 바꾸는 과정에서 void값을 255로 지정했는데, 여기서 문제가 생겨서 11로 변경해서 에러 해결했습니다.

0~10까지는 class이고 11은 void 입니다!