.
├── convert.py
├── datasets
│   └── gen_datasets
│            ├── train
│            │   ├── images
│            │   └── masks
│            └── val
│                ├── images
|                └── masks
├── eval_onnxruntime.py
├── loss.py
├── pytorch_unet.py
├── requirements.txt
└── train_self.py