/ynet

Primary LanguagePython

Y-Net

To train a Y-Net model, run python train_ynet.py. To train a U-Net model, run python train_unet.py.

You can evaluate trained models with python eval_model.py.