junkim1209 commited on
Commit
bd1c056
ยท
verified ยท
1 Parent(s): eda52d5

Add model card with medical disclaimer

Browse files
Files changed (1) hide show
  1. README.md +41 -0
README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-nc-4.0
3
+ tags:
4
+ - medical-imaging
5
+ - dermatology
6
+ - melanoma
7
+ - isic
8
+ - education
9
+ ---
10
+
11
+ # ISIC ๋ฉœ๋ผ๋…ธ๋งˆ ๋ถ„๋ฅ˜ 5-Model
12
+
13
+ ๊ทธ๋กœ์„ผ๋”•(GrowthEndix) AI ๊ต์œก๊ณผ์ • โ€” ISIC 2019+2020 ํ•™์Šต ๋ชจ๋ธ 5์ข…
14
+
15
+ ## โš ๏ธ ์˜๋ฃŒ ๋””์Šคํด๋ ˆ์ด๋จธ
16
+
17
+ ๋ณธ ๋ชจ๋ธ์€ **๊ต์œก ๋ฐ ์—ฐ๊ตฌ ๋ชฉ์ **์œผ๋กœ๋งŒ ๊ณต๊ฐœ๋ฉ๋‹ˆ๋‹ค.
18
+ - ์‹ค์ œ ์˜๋ฃŒ ์ง„๋‹จ์— ์‚ฌ์šฉ ๋ถˆ๊ฐ€
19
+ - ํ”ผ๋ถ€ ์ด์ƒ ์˜์‹ฌ ์‹œ ๋ฐ˜๋“œ์‹œ ํ”ผ๋ถ€๊ณผ ์ „๋ฌธ์˜ ์ง„๋ฃŒ
20
+ - ISIC ๋ฐ์ดํ„ฐ์…‹์˜ ์ธ๊ตฌ ํŽธํ–ฅ(์ฃผ๋กœ ๋ฐฑ์ธ ํ™˜์ž)์œผ๋กœ ํ•œ๊ตญ์ธ ํ”ผ๋ถ€ํ†ค ์ •ํ™•๋„ ๋ฏธ๋ณด์žฅ
21
+
22
+ ## ๋ชจ๋ธ ์‚ฌ์–‘
23
+
24
+ | ๋ชจ๋ธ | ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ | ์ž…๋ ฅ ํ•ด์ƒ๋„ | val_auc |
25
+ |---|---|---|---|
26
+ | efficientnet_b3 | torchvision | 300x300 | 0.9288 |
27
+ | efficientnet_b5 | timm | 456x456 | 0.9246 |
28
+ | convnext_tiny | torchvision | 224x224 | 0.9164 |
29
+ | convnext_base | timm | 224x224 | 0.9178 |
30
+ | swin_base | timm (window12_384) | 384x384 | 0.9195 |
31
+
32
+ ## ์‚ฌ์šฉ๋ฒ•
33
+
34
+ ```python
35
+ from huggingface_hub import hf_hub_download
36
+ path = hf_hub_download(repo_id="junkim1209/isic-models", filename="efficientnet_b3.pth")
37
+ ```
38
+
39
+ - ์ €์žฅ ํ˜•์‹: ์ฒดํฌํฌ์ธํŠธ wrapping (`state['model_state_dict']`)
40
+ - ์ „์ฒ˜๋ฆฌ: ImageNet ์ •๊ทœํ™”, ๋ชจ๋ธ๋ณ„ ํ•ด์ƒ๋„๋กœ resize
41
+ - Head: ์ด์ง„ ๋ถ„๋ฅ˜ (sigmoid ์ ์šฉ ์‹œ ๋ฉœ๋ผ๋…ธ๋งˆ ํ™•๋ฅ )