未验证 提交 d95203ea 编写于 作者: J Jason 提交者: GitHub

Merge pull request #84 from PaddlePaddle/develop_interpret

resume hub
import paddlex import paddlex
#import paddlehub as hub import paddlehub as hub
import os import os
import os.path as osp import os.path as osp
...@@ -85,53 +85,49 @@ def get_pretrain_weights(flag, model_type, backbone, save_dir): ...@@ -85,53 +85,49 @@ def get_pretrain_weights(flag, model_type, backbone, save_dir):
backbone = 'DetResNet50' backbone = 'DetResNet50'
assert backbone in image_pretrain, "There is not ImageNet pretrain weights for {}, you may try COCO.".format( assert backbone in image_pretrain, "There is not ImageNet pretrain weights for {}, you may try COCO.".format(
backbone) backbone)
url = image_pretrain[backbone] # url = image_pretrain[backbone]
fname = osp.split(url)[-1].split('.')[0] # fname = osp.split(url)[-1].split('.')[0]
paddlex.utils.download_and_decompress(url, path=new_save_dir) # paddlex.utils.download_and_decompress(url, path=new_save_dir)
return osp.join(new_save_dir, fname) # return osp.join(new_save_dir, fname)
# try: try:
# hub.download(backbone, save_path=new_save_dir) hub.download(backbone, save_path=new_save_dir)
# except Exception as e: except Exception as e:
# if isinstance(e, hub.ResourceNotFoundError): if isinstance(e, hub.ResourceNotFoundError):
# raise Exception( raise Exception("Resource for backbone {} not found".format(
# "Resource for backbone {} not found".format(backbone)) backbone))
# elif isinstance(e, hub.ServerConnectionError): elif isinstance(e, hub.ServerConnectionError):
# raise Exception( raise Exception(
# "Cannot get reource for backbone {}, please check your internet connecgtion" "Cannot get reource for backbone {}, please check your internet connecgtion"
# .format(backbone)) .format(backbone))
# else: else:
# raise Exception( raise Exception(
# "Unexpected error, please make sure paddlehub >= 1.6.2") "Unexpected error, please make sure paddlehub >= 1.6.2")
# return osp.join(new_save_dir, backbone) return osp.join(new_save_dir, backbone)
elif flag == 'COCO': elif flag == 'COCO':
new_save_dir = save_dir new_save_dir = save_dir
if hasattr(paddlex, 'pretrain_dir'): if hasattr(paddlex, 'pretrain_dir'):
new_save_dir = paddlex.pretrain_dir new_save_dir = paddlex.pretrain_dir
url = coco_pretrain[backbone] url = coco_pretrain[backbone]
fname = osp.split(url)[-1].split('.')[0] fname = osp.split(url)[-1].split('.')[0]
paddlex.utils.download_and_decompress(url, path=new_save_dir) # paddlex.utils.download_and_decompress(url, path=new_save_dir)
return osp.join(new_save_dir, fname) # return osp.join(new_save_dir, fname)
# new_save_dir = save_dir assert backbone in coco_pretrain, "There is not COCO pretrain weights for {}, you may try ImageNet.".format(
# if hasattr(paddlex, 'pretrain_dir'): backbone)
# new_save_dir = paddlex.pretrain_dir try:
# assert backbone in coco_pretrain, "There is not COCO pretrain weights for {}, you may try ImageNet.".format( hub.download(backbone, save_path=new_save_dir)
# backbone) except Exception as e:
# try: if isinstance(hub.ResourceNotFoundError):
# hub.download(backbone, save_path=new_save_dir) raise Exception("Resource for backbone {} not found".format(
# except Exception as e: backbone))
# if isinstance(hub.ResourceNotFoundError): elif isinstance(hub.ServerConnectionError):
# raise Exception( raise Exception(
# "Resource for backbone {} not found".format(backbone)) "Cannot get reource for backbone {}, please check your internet connecgtion"
# elif isinstance(hub.ServerConnectionError): .format(backbone))
# raise Exception( else:
# "Cannot get reource for backbone {}, please check your internet connecgtion" raise Exception(
# .format(backbone)) "Unexpected error, please make sure paddlehub >= 1.6.2")
# else: return osp.join(new_save_dir, backbone)
# raise Exception(
# "Unexpected error, please make sure paddlehub >= 1.6.2")
# return osp.join(new_save_dir, backbone)
else: else:
raise Exception( raise Exception(
"pretrain_weights need to be defined as directory path or `IMAGENET` or 'COCO' (download pretrain weights automatically)." "pretrain_weights need to be defined as directory path or `IMAGENET` or 'COCO' (download pretrain weights automatically)."
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册