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 ์ ์šฉ ์‹œ ๋ฉœ๋ผ๋…ธ๋งˆ ํ™•๋ฅ )