提交 53e9d503 编写于 作者: W wuzewu
...@@ -66,9 +66,6 @@ MODEL: ...@@ -66,9 +66,6 @@ MODEL:
DEFAULT_NORM_TYPE: "bn" DEFAULT_NORM_TYPE: "bn"
DEEPLAB: DEEPLAB:
BACKBONE: "xception_65" BACKBONE: "xception_65"
TRAIN:
PRETRAINED_MODEL_DIR: "./pretrained_model/deeplabv3p_xception65_bn_pet/"
# 其他配置 # 其他配置
TRAIN_CROP_SIZE: (512, 512) TRAIN_CROP_SIZE: (512, 512)
...@@ -78,6 +75,7 @@ AUG: ...@@ -78,6 +75,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/deeplabv3p_xception65_bn_pet/"
MODEL_SAVE_DIR: "./finetune/deeplabv3p_xception65_bn_pet/" MODEL_SAVE_DIR: "./finetune/deeplabv3p_xception65_bn_pet/"
SNAPSHOT_EPOCH: 10 SNAPSHOT_EPOCH: 10
TEST: TEST:
......
...@@ -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:
......
...@@ -64,9 +64,6 @@ DATASET: ...@@ -64,9 +64,6 @@ DATASET:
MODEL: MODEL:
MODEL_NAME: "unet" MODEL_NAME: "unet"
DEFAULT_NORM_TYPE: "bn" DEFAULT_NORM_TYPE: "bn"
TRAIN:
PRETRAINED_MODEL_DIR: "./pretrained_model/unet_bn_coco/"
# 其他配置 # 其他配置
TRAIN_CROP_SIZE: (512, 512) TRAIN_CROP_SIZE: (512, 512)
...@@ -76,6 +73,7 @@ AUG: ...@@ -76,6 +73,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/unet_bn_coco/"
MODEL_SAVE_DIR: "./saved_model/unet_pet/" MODEL_SAVE_DIR: "./saved_model/unet_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.
先完成此消息的编辑!
想要评论请 注册