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.
Amir a96d0ad51d first run 2 years ago
datasets delete python cache 2 years ago
models delete python cache 2 years ago
utils first run 2 years ago
.gitignore gitignore 2 years ago
README.md Update README.md 3 years ago
opt.py first run 2 years ago
test.py Add files via upload 3 years ago
train.py first run 2 years ago

README.md

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

Requirements

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

Data

$ data
train
├── Images
├── Masks
valid
├── Images
├── Masks
test
├── 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