isic-models / README.md
junkim1209's picture
Add model card with medical disclaimer
bd1c056 verified
---
license: cc-by-nc-4.0
tags:
- medical-imaging
- dermatology
- melanoma
- isic
- education
---
# 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 |
## ์‚ฌ์šฉ๋ฒ•
```python
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 ์ ์šฉ ์‹œ ๋ฉœ๋ผ๋…ธ๋งˆ ํ™•๋ฅ )