Segment & Classify for Treponema pallidum
매독균 분류 프로젝트
- PyTorch Lightning을 이용해 모델 학습 코드 구현
- Confusion matrix 시각화 및 Metric logging
- OpenCV 이용한 전처리 (Contrast Enhancement, Sharpening)
- SegGPT모델 사용한 자동 Segment 파이프라인 구축
- resize의 영향 줄이기 위해 원본 이미지 split하여 예측하고 이를 평균내어 결과 예측에 사용
- 원본 이미지와 segment mask의 overlay 픽셀의 intensity값의 분포 파악하여 연관성 확인
- noramlized intensity histogram distribution을 model의 fc layer에 concat하여 모델 구조 변경하여 성능 개선