|
12345678910111213141516171819202122232425262728 |
- # Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation
-
- ## Requirements
-
- * torch
- * torchvision
- * scipy
- * PIL
- * numpy
- * tqdm
-
- ### Data
-
-
-
-
-
- ### 1. Training
-
- ```bash
- 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```
- ```
|