GitHub - gbvo/Enhanced-U-Net

Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation

$ data
train
├── Images
├── Masks
valid
├── Images
├── Masks
test
├── Images
├── Masks

python train.py  --mode train  --train_data_dir /path-to-train_data  --valid_data_dir  /path-to-valid_data

python test.py  --mode test  --load_ckpt checkpoint --test_data_dir  /path-to-test_data

Name		Name	Last commit message	Last commit date
Latest commit History 31 Commits
datasets		datasets
models		models
utils		utils
README.md		README.md
opt.py		opt.py
test.py		test.py
train.py		train.py