File size: 1,246 Bytes
bd1c056 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | ---
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 ์ ์ฉ ์ ๋ฉ๋ผ๋
ธ๋ง ํ๋ฅ )
|