Unet based on Wavelet coefficients for segmentation
Dataset format:
Train Folder-----
img----
img1.png
img2.png
.......
label---
label1.png
label2.png
.......
Test Folder-----
img----
img1.png
img2.png
.......
label---
label1.png
label2.png
.......
Running Command for train.py :
python3 train.py --train_dataset 'your directory' --val_dataset 'your directory' --model_name 'brainus_db2_lvl2' --checkpoint_path 'chk/brainus__lvl2'
Wavelets and Filter Banks, JHU Fall 2019.