def download_model(self): """Download model from url.""" cache_dir = Test.cache_dir if not os.path.exists(cache_dir): os.makedirs(cache_dir) url = self.url if url.startswith(r'module://'): return self.download_from_module() k = url.rfind('/') fname = self.url[k + 1:] dir_name = fname + "_dir" ftype = None if url.endswith(".tar.gz") or url.endswith(".tgz"): ftype = 'tgz' dir_name = fname.replace(".tar.gz", "").replace(".tgz", "") elif url.endswith('.zip'): ftype = 'zip' dir_name = fname.replace(".zip", "") dir_name = os.path.join(cache_dir, dir_name) os.makedirs(dir_name, exist_ok=True) fpath = os.path.join(dir_name, fname) if not os.path.exists(fpath): utils.get_url(url, fpath) model_path = os.path.join(dir_name, self.local) if not os.path.exists(model_path) or self.local == ".": if ftype == 'tgz': tar = tarfile.open(fpath) tar.extractall(dir_name) tar.close() elif ftype == 'zip': zip_ref = zipfile.ZipFile(fpath, 'r') zip_ref.extractall(dir_name) zip_ref.close() return fpath, dir_name
def download_tflite(url, dest, verbose=True): """ Downloads a model from tfhub. The function assumes the format is `.tflite`. """ if not os.path.exists(dest): os.makedirs(dest) fpath = os.path.join(dest, "model.tflite") if not os.path.exists(fpath): from tf2onnx import utils if verbose: print("Download %r." % fpath) utils.get_url(url, fpath) return fpath
def download_model(url, dest, verbose=True): """ Downloads a model from tfhub and unzips it. The function assumes the format is `.tar.gz`. """ if not os.path.exists(dest): os.makedirs(dest) fpath = os.path.join(dest, "model.tar.gz") if not os.path.exists(fpath): if verbose: print("Download %r." % fpath) utils.get_url(url, fpath) tname = os.path.join(dest, "model_path") if not os.path.exists(tname): if verbose: print("Untar %r." % tname) tar = tarfile.open(fpath) tar.extractall(tname) tar.close() return fpath, tname