Pārlūkot izejas kodu

use hub to download hr pretrained weights

FlyingQianMM 5 gadi atpakaļ
vecāks
revīzija
336a4fef3f
1 mainītis faili ar 4 papildinājumiem un 5 dzēšanām
  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: