rucv
3 years ago
No account linked to committer's email address
1 changed files with
3 additions and
3 deletions
-
README.md
|
|
@@ -1,6 +1,6 @@ |
|
|
|
# Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation |
|
|
|
## Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation |
|
|
|
|
|
|
|
## Requirements |
|
|
|
### Requirements |
|
|
|
|
|
|
|
* torch |
|
|
|
* torchvision |
|
|
@@ -29,7 +29,7 @@ test |
|
|
|
python train.py --mode train --train_data_dir /path-to-train_data --valid_data_dir /path-to-valid_data |
|
|
|
``` |
|
|
|
|
|
|
|
#### 2. Testing |
|
|
|
### 2. Testing |
|
|
|
|
|
|
|
```bash |
|
|
|
python test.py --mode test --load_ckpt checkpoint --test_data_dir /path-to-test_data |