Vision/Segmentation (1) 썸네일형 리스트형 [detectron2] unet 적용 시 num_class 이슈 및 해결방법 Facebook에서 개발한 detectron2의 U-net Implement 라이브러리(아래 링크)를 활용하는 과정에서 문제가 발생 꽤 많은 삽질(약 이틀)을 하여 기록하기 위해 남겨둠 https://github.com/zongyue-lu/pytorch-unet-family 1. 이슈 요약 이슈는 크게 세가지로 나누어 볼 수 있음 Semantic Segmentation용 데이터 생성 및 로드 관련 이슈 NUM_CLASS 이슈 Cross Entropy Loss 이슈 2. 이슈 상세 1) 데이터 생성 과정 - 기존 COCO object detection용 데이터를 SS용 데이터로 변환하는 과정에서, 각 픽셀에 label 정보를 입력하고 png 형태로 저장 - label 정보는 1부터 시작하여 C(Num or.. 이전 1 다음