You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
rucv 5186759ecf
Update README.md
2 years ago
datasets Add files via upload 2 years ago
models Add files via upload 2 years ago
utils Add files via upload 2 years ago
README.md Update README.md 2 years ago
opt.py Add files via upload 2 years ago
test.py Add files via upload 2 years ago
train.py Add files via upload 2 years ago

README.md

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

Requirements

  • torch
  • torchvision
  • scipy
  • PIL
  • numpy
  • tqdm

Data

train \ Images \ Masks \ test \ Images \ Masks \ Valid \ Images \ Masks

1. Training

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

2. Testing

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