소스 검색

use hub to download hr pretrained weights

FlyingQianMM 5 년 전
부모
커밋
336a4fef3f
1개의 변경된 파일4개의 추가작업 그리고 5개의 파일을 삭제
  1. 4 5
      paddlex/cv/models/utils/pretrain_weights.py

+ 4 - 5
paddlex/cv/models/utils/pretrain_weights.py

@@ -99,11 +99,10 @@ def get_pretrain_weights(flag, model_type, backbone, save_dir):
                 backbone = 'DetResNet50'
         assert backbone in image_pretrain, "There is not ImageNet pretrain weights for {}, you may try COCO.".format(
             backbone)
-        if backbone.startswith("HRNet"):
-            url = image_pretrain[backbone]
-            fname = osp.split(url)[-1].split('.')[0]
-            paddlex.utils.download_and_decompress(url, path=new_save_dir)
-            return osp.join(new_save_dir, fname)
+        #        url = image_pretrain[backbone]
+        #        fname = osp.split(url)[-1].split('.')[0]
+        #        paddlex.utils.download_and_decompress(url, path=new_save_dir)
+        #        return osp.join(new_save_dir, fname)
         try:
             hub.download(backbone, save_path=new_save_dir)
         except Exception as e: