Deep Active Contours Model(ACM) for Buildings Segmentations

Deep snake algorithm for 2D images based on - [ICLR2020 paper(revisiting the results)](https://arxiv.org/abs/1912.00367) Architecture based on [Hardnet85](https://arxiv.org/abs/1909.00948) Data and weights = (https://drive.google.com/drive/folders/1fBSjPse3d8geV_iI3-PXV3x2qmLoUnzL?usp=sharing)

To train a segmentation model :

python train_seg.py -bs 50 -WD 0.00005 -D_rate 3000 -task bing -opt sgd -lr 0.02 -nW 8

To train a ACM model :

python train.py -bs 25 -WD 0.00005 -D_rate 30 -it 2

To eval a segmentation model :

python eval.py -task viah -nP 100 -it 3 -a 0.4

To eval a ACM model :

 python eval_seg.py -task bing -nP 24 -it 2 -a 0.4

Name		Name	Last commit message	Last commit date
Latest commit History 12 Commits
Data		Data
hd5_maker		hd5_maker
loader		loader
models		models
out		out
pics		pics
results		results
utils		utils
.gitattributes		.gitattributes
README.md		README.md
eval.py		eval.py
eval_seg.py		eval_seg.py
requirements.txt		requirements.txt
train.py		train.py
train_seg.py		train_seg.py

talshaharabany/DeepACM2D