[제안] Train & Validation & Test prediction 시각화(mmsegmentation 용)

[Devlee247]

개요 및 목적

Train & Validation & Test prediction 시각화(mmsegmentation 용)

작업 상세 내용

• [x] Train 시각화
• [x] Validation 시각화
• [x] Test 시각화

참고 문서

[agwmon]

ipynb 형태로 inference 과정까지 진행하는 기능 추가하였습니다. utils/pred_vis에 이전처럼 모델 config & checkpoint만 입력하시고, 순서대로 진행하시면 됩니다. train inference 단계에서 swinL 기준 5~6분 소요됩니다.

vis_tool(mode, index) 형태로 사용하시면 됩니다. (mode: train, test, val)

pixel_score는 제가 임의로 만든 background를 제외한 정답 픽셀 비율을 의미하는 성능 지표입니다. mask 덧붙여진 이미지가 보기 불편하시면 vis_tool 함수에서 opacity(투명도) 조정하시면 됩니다!