Name	Size	Mode	Actions
download.cpython-36.pyc	6506	0644	editdlrm
imagenet_trainer_test_utils.cpython-36.pyc	4089	0644	editdlrm
resnet.cpython-36.pyc	6369	0644	editdlrm
resnet_test.cpython-36.pyc	2391	0644	editdlrm
shufflenet.cpython-36.pyc	4416	0644	editdlrm
shufflenet_test.cpython-36.pyc	2451	0644	editdlrm
__init__.cpython-36.pyc	156	0644	editdlrm
__sym_init__.cpython-36.pyc	615	0644	editdlrm

3 EgC�@s@
dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd
lZddlmZ
ddlm Z m Z
dd
ljZ dZdZdd�Zejeje�
dd �Zd d�Zd'd d�
Zdd�Zdd�Zdd�ZGdd�d�Zedk�
rr r r� downloadModelSsN

































rYc
Cs&d
g
}
||
krdStj
d|�s"dSdS)N�__init__Fz^[/0-9a-zA-Z_-]+$T)�re�match)r� invalid_namesr r r�validModelName�s





r^c@s6eZ
dZd dd�
Zdd�Zdd�Zdd �Zd d�ZdS)�ModelDownloader� CAFFE2_MODELScCs |
|_dS)
N)
�model_env_name)�selfrar r rrZ�s
zModelDownloader.__init__cCs:tj
jtjd
d��
}tj|jtj
j|d��}tj
j||
�S)NZCAFFE2_HOMEz ~/.caffe2�models)rr� expanduser�getenvrar)rbrVZcaffe2_homeZ models_dirr r r� _model_dir�s


zModelDownloader._model_dircCs�|j|
�
}t
jj|�
st�
t
j|�

xrdD]j}t|
|�}t
jj||�}yt||dd�
Wq,t k rx


t||�
Yq,t k r�


t|�

�Yq,Xq,WdS)N�predict_net.pb�init_net.pb�value_info.jsonF)
r9)rgrhri)rfrrrM�AssertionErrorrTrBrr?� TypeErrorr8r)rbrV� model_dirrHr(�destr r r� _download�s










zModelDownloader._downloadcCs$
d
}|j|
�
}t
jj|�
s$|j|
�

t
jj|d�}|d|d7}tj�}t|d��(}|j |j ��
}|dt|�
d7}WdQRX|
|_t
jj|d�}|d|d7}tj�} t|d��(}| j |j ��
}|d t|�
d7}WdQRX|
d | _tt
jj|d��
�}t j|�
} WdQRX| || |fS)Nzget_c2_model debug: zpredict_net.pbzc2_predict_pb path: �
�rbz c2_predict_pb ParseFromString = zinit_net.pbzc2_init_pb path: zc2_init_pb ParseFromString = �_initzvalue_info.json)rfrrrMrnrrZNetDefr5ZParseFromStringr6r"r�json�load)rb� model_nameZ debug_strrlZ c2_predict_pbZc2_predict_netrHZlen_readZ c2_init_pbZc2_init_net� value_infor r r�get_c2_model_dbg�s*

















z ModelDownloader.get_c2_model_dbgcCs|j|
�
\}}}}|||fS)
N)
rv)rbrtZinit_netZpredict_netru�
_r r r�get_c2_model�s

zModelDownloader.get_c2_modelN)
r`)�__name__� __module__�__qualname__rZrfrnrvrxr r r rr_�s  r_�__main__z&Download or install pretrained models.)
�descriptionrV�
+zModel to download/install.)�nargs�helpz-iz --install� store_truezInstall the model.)�actionr�z-fz--forcezForce a download/installation.z'{}' is not a valid model name.)
T)"�argparserrrr[rrZcaffe2.protor�urllib.errorrr�urllib.request�requestr/rArr�SIGINTrr'r?rBrYr^r_ry�ArgumentParser�parser�add_argument� parse_argsrWrVrr.r r r r�sB






  1=