浏览代码

Add P2V pretrained weights

Bobholamovic 2 年之前
父节点
当前提交
cb8135c7f4
共有 1 个文件被更改,包括 3 次插入0 次删除
  1. 3 0
      paddlers/utils/checkpoint.py

+ 3 - 0
paddlers/utils/checkpoint.py

@@ -30,6 +30,7 @@ cd_pretrain_weights_dict = {
     'FCSiamConc': ['LEVIRCD'],
     'FCSiamConc': ['LEVIRCD'],
     'FCSiamDiff': ['LEVIRCD'],
     'FCSiamDiff': ['LEVIRCD'],
     'FCCDN': ['LEVIRCD'],
     'FCCDN': ['LEVIRCD'],
+    'P2V': ['LEVIRCD'],
     'SNUNet': ['LEVIRCD'],
     'SNUNet': ['LEVIRCD'],
     'STANet': ['LEVIRCD']
     'STANet': ['LEVIRCD']
 }
 }
@@ -442,6 +443,8 @@ levircd_weights = {
     'https://paddlers.bj.bcebos.com/pretrained/cd/levircd/weights/fc_siam_diff_levircd.pdparams',
     'https://paddlers.bj.bcebos.com/pretrained/cd/levircd/weights/fc_siam_diff_levircd.pdparams',
     'FCCDN_LEVIRCD':
     'FCCDN_LEVIRCD':
     'https://paddlers.bj.bcebos.com/pretrained/cd/levircd/weights/fccdn_levircd.pdparams',
     'https://paddlers.bj.bcebos.com/pretrained/cd/levircd/weights/fccdn_levircd.pdparams',
+    'P2V_LEVIRCD':
+    'https://paddlers.bj.bcebos.com/pretrained/cd/levircd/weights/p2v_levircd.pdparams',
     'SNUNet_LEVIRCD':
     'SNUNet_LEVIRCD':
     'https://paddlers.bj.bcebos.com/pretrained/cd/levircd/weights/snunet_levircd.pdparams',
     'https://paddlers.bj.bcebos.com/pretrained/cd/levircd/weights/snunet_levircd.pdparams',
     'STANet_LEVIRCD':
     'STANet_LEVIRCD':