浏览代码

remove imagenet weights of deeplabv3p

will-jl944 4 年之前
父节点
当前提交
e7ad5f083c
共有 2 个文件被更改,包括 2 次插入7 次删除
  1. 0 1
      dygraph/examples/meter_reader/train_segmentation.py
  2. 2 6
      dygraph/paddlex/utils/checkpoint.py

+ 0 - 1
dygraph/examples/meter_reader/train_segmentation.py

@@ -48,7 +48,6 @@ model.train(
     num_epochs=20,
     train_dataset=train_dataset,
     train_batch_size=4,
-    pretrain_weights='IMAGENET',
     eval_dataset=eval_dataset,
     learning_rate=0.1,
     save_dir='output/deeplabv3p_r50vd')

+ 2 - 6
dygraph/paddlex/utils/checkpoint.py

@@ -20,7 +20,7 @@ from .download import download_and_decompress
 
 seg_pretrain_weights_dict = {
     'UNet': ['CITYSCAPES'],
-    'DeepLabV3P': ['CITYSCAPES', 'PascalVOC', 'IMAGENET'],
+    'DeepLabV3P': ['CITYSCAPES', 'PascalVOC'],
     'FastSCNN': ['CITYSCAPES'],
     'HRNet': ['CITYSCAPES', 'PascalVOC'],
     'BiSeNetV2': ['CITYSCAPES']
@@ -254,11 +254,7 @@ imagenet_weights = {
     'MaskRCNN_ResNet101_fpn_IMAGENET':
     'https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams',
     'MaskRCNN_ResNet101_vd_fpn_IMAGENET':
-    'https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams',
-    'DeepLabV3P_ResNet50_vd_IMAGENET':
-    'https://bj.bcebos.com/paddleseg/dygraph/resnet50_vd_ssld.tar.gz',
-    'DeepLabV3P_ResNet101_vd_IMAGENET':
-    'https://bj.bcebos.com/paddleseg/dygraph/resnet101_vd_ssld.tar.gz'
+    'https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams'
 }
 
 pascalvoc_weights = {