未验证 提交 f5605328 编写于 作者: W whs 提交者: GitHub

Fix doc and train.py of icnet. (#2832)

上级 cf2f5939
......@@ -66,7 +66,7 @@ Iter[0]; train loss: 2.338; sub4_loss: 3.367; sub24_loss: 4.120; sub124_loss: 0.
### 测试
执行以下命令在`Cityscape`测试数据集上进行测试:
```
python eval.py --model_path="./cnkpnt/100" --use_gpu=True
python eval.py --model_path="./chkpnt/100" --use_gpu=True
```
需要通过选项`--model_path`指定模型文件。
测试脚本的输出的评估指标为[mean IoU]()。
......@@ -75,7 +75,7 @@ python eval.py --model_path="./cnkpnt/100" --use_gpu=True
执行以下命令对指定的数据进行预测:
```
python infer.py \
--model_path="./cnkpnt/100" \
--model_path="./chkpnt/100" \
--images_path="./data/cityscape/" \
--images_list="./data/cityscape/infer.list"
```
......
......@@ -35,9 +35,11 @@ LAMBDA2 = 0.4
LAMBDA3 = 1.0
LEARNING_RATE = 0.003
POWER = 0.9
LOG_PERIOD = 100
CHECKPOINT_PERIOD = 100
TOTAL_STEP = 100
LOG_PERIOD = 1
CHECKPOINT_PERIOD = 1000
TOTAL_STEP = 60000
if 'ce_mode' in os.environ:
TOTAL_STEP = 100
no_grad_set = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册