junkim1209/isic-models

ISIC 멜라노마 분류 5-Model

그로센딕(GrowthEndix) AI 교육과정 — ISIC 2019+2020 학습 모델 5종

⚠️ 의료 디스클레이머

본 모델은 교육 및 연구 목적으로만 공개됩니다.

• 실제 의료 진단에 사용 불가
• 피부 이상 의심 시 반드시 피부과 전문의 진료
• ISIC 데이터셋의 인구 편향(주로 백인 환자)으로 한국인 피부톤 정확도 미보장

모델 사양

모델	라이브러리	입력 해상도	val_auc
efficientnet_b3	torchvision	300x300	0.9288
efficientnet_b5	timm	456x456	0.9246
convnext_tiny	torchvision	224x224	0.9164
convnext_base	timm	224x224	0.9178
swin_base	timm (window12_384)	384x384	0.9195

사용법

from huggingface_hub import hf_hub_download
path = hf_hub_download(repo_id="junkim1209/isic-models", filename="efficientnet_b3.pth")

• 저장 형식: 체크포인트 wrapping (state['model_state_dict'])
• 전처리: ImageNet 정규화, 모델별 해상도로 resize
• Head: 이진 분류 (sigmoid 적용 시 멜라노마 확률)