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