| 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 ์ ์ฉ ์ ๋ฉ๋ผ๋ ธ๋ง ํ๋ฅ ) | |