Landscape Segmentation with Deeplab V3
์ด ํ๋ก์ ํธ๋ ํญ๊ณต์ฌ์ง ๊ธฐ๋ฐ์ผ๋ก ์งํ์ ๋ถ๋ฅํ๋ ํ๋ก์ ํธ ์ ๋๋ค.
- Language: Python 3.8
- Libraries: PyTorch, OpenCV, NumPy, pandas, PIL
- Model: Deeplab V3+
- Tool: VS code, Git
- Segmentation ๊ธฐ๋ฐ ์ปค์คํ ๋ฐ์ดํฐ์
- Segmentation ๋ง์คํฌ ๋ฐ ์ด๋ฏธ์ง ๋ฐ์ดํฐ ์ฆ๊ฐ (Random_crop,Resize,Add noise ETC)
- Focal loss ๋ฐ soft_dicel loss ๊ธฐ๋ฐ์ Loss ๊ณ์ฐ ๋ฐ Evaluation
- ์ถ๋ก ๊ฒฐ๊ณผ๋ฅผ ์ด๋ฏธ์ง๋ก ์๊ฐํ
project/
โโโ data/ # ์ด๋ฏธ์ง ๋ฐ ๋ง์คํฌ ๋ฐ์ดํฐ์
โโโ Modules/ # ๋ชจ๋ธ ๋ฐ ํ์ต ๊ด๋ จ ๋ชจ๋
โ โโโ DeepLabV3Plus.py # DeepLab V3+ ๋ชจ๋ธ ๊ตฌํ
โ โโโ Trainer.py # ๋ชจ๋ธ ํ์ต ๋ชจ๋
โ โโโ Configuration.py # ์ค์ ํด๋์ค
โ โโโ dataset.py # ์ด๋ฏธ์ง ์ฆ๊ฐ ๋ฐ ์ปค์คํ
๋ฐ์ดํฐ์
โ โโโ evaluation.py # ์ฑ๋ฅ ํ๊ฐ ํจ์
โ โโโ xception.py # Xception ๋ชจ๋ธ
โ (์ถ์ฒ: https://github.com/Cadene/pretrained-models.pytorch/blob/master/pretrainedmodels/models/xception.py)
โโโ image_segmentation_project.ipynb # ์ ์ฒด ์คํ Jupyter ๋
ธํธ๋ถ