提交 b16588a8 编写于 作者: J jiangjiajun

add alexnet

上级 0f85534c
...@@ -102,11 +102,11 @@ def get_pretrain_weights(flag, model_type, backbone, save_dir): ...@@ -102,11 +102,11 @@ def get_pretrain_weights(flag, model_type, backbone, save_dir):
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)
if backbone == 'AlexNet': # if backbone == 'AlexNet':
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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册