rucv
4 years ago
1 changed files with 3 additions and 3 deletions
+ 3
- 3
README.md
View File
| # Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation | |||||
| ## Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation | |||||
| ## Requirements | |||||
| ### Requirements | |||||
| * torch | * torch | ||||
| * torchvision | * torchvision | ||||
| python train.py --mode train --train_data_dir /path-to-train_data --valid_data_dir /path-to-valid_data | python train.py --mode train --train_data_dir /path-to-train_data --valid_data_dir /path-to-valid_data | ||||
| ``` | ``` | ||||
| #### 2. Testing | |||||
| ### 2. Testing | |||||
| ```bash | ```bash | ||||
| python test.py --mode test --load_ckpt checkpoint --test_data_dir /path-to-test_data | python test.py --mode test --load_ckpt checkpoint --test_data_dir /path-to-test_data |
Loading…