未验证 提交 71950612 编写于 作者: W wuzewu 提交者: GitHub

Update finetune_icnet.md

上级 1e6eac0a
...@@ -65,9 +65,6 @@ MODEL: ...@@ -65,9 +65,6 @@ MODEL:
MODEL_NAME: "icnet" MODEL_NAME: "icnet"
DEFAULT_NORM_TYPE: "bn" DEFAULT_NORM_TYPE: "bn"
MULTI_LOSS_WEIGHT: "[1.0, 0.4, 0.16]" MULTI_LOSS_WEIGHT: "[1.0, 0.4, 0.16]"
TRAIN:
PRETRAINED_MODEL_DIR: "./pretrained_model/icnet_bn_cityscapes/"
# 其他配置 # 其他配置
TRAIN_CROP_SIZE: (512, 512) TRAIN_CROP_SIZE: (512, 512)
...@@ -77,6 +74,7 @@ AUG: ...@@ -77,6 +74,7 @@ AUG:
FIX_RESIZE_SIZE: (512, 512) FIX_RESIZE_SIZE: (512, 512)
BATCH_SIZE: 4 BATCH_SIZE: 4
TRAIN: TRAIN:
PRETRAINED_MODEL_DIR: "./pretrained_model/icnet_bn_cityscapes/"
MODEL_SAVE_DIR: "./saved_model/icnet_pet/" MODEL_SAVE_DIR: "./saved_model/icnet_pet/"
SNAPSHOT_EPOCH: 10 SNAPSHOT_EPOCH: 10
TEST: TEST:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册