Delete utils

utils/0

@@ -1 +0,0 @@
-y

utils/__init__.py

@@ -1,2 +0,0 @@
-from .download import download_file
-from .weights_urls import get_model_url

utils/download.py

@@ -1,45 +0,0 @@
-import os
-import requests
-import zipfile
-
-from tqdm import tqdm
-
-
-def download_file(url: str, save_dir='./', overwrite=False, unzip=True):
-    os.makedirs(save_dir, exist_ok=True)
-    file_name = url.split('/')[-1]
-    file_path = os.path.join(save_dir, file_name)
-
-    if os.path.exists(file_path) and not overwrite:
-        pass
-    else:
-        print('Downloading file {} from {}...'.format(file_path, url))
-
-        r = requests.get(url, stream=True)
-        print(r.status_code)
-        if r.status_code != 200:
-            raise RuntimeError('Failed downloading url {}!'.format(url))
-        total_length = r.headers.get('content-length')
-        with open(file_path, 'wb') as f:
-            if total_length is None:  # no content length header
-                for chunk in r.iter_content(chunk_size=1024):
-                    if chunk:  # filter out keep-alive new chunks
-                        f.write(chunk)
-            else:
-                total_length = int(total_length)
-                print('file length: ', int(total_length / 1024. + 0.5))
-                for chunk in tqdm(r.iter_content(chunk_size=1024),
-                                  total=int(total_length / 1024. + 0.5),
-                                  unit='KB',
-                                  unit_scale=False,
-                                  dynamic_ncols=True):
-                    f.write(chunk)
-    if unzip and file_path.endswith('.zip'):
-        save_dir = file_path.split('.')[0]
-        if os.path.isdir(save_dir) and os.path.exists(save_dir):
-            pass
-        else:
-            with zipfile.ZipFile(file_path, 'r') as zip_ref:
-                zip_ref.extractall(save_dir)
-
-    return save_dir, file_path

utils/weights_urls.py

@@ -1,22 +0,0 @@
-WEIGHT_URLS = {
-    'buffalo_l':
-    'https://github.com/justld/dofaker/releases/download/v0.1/buffalo_l.zip',
-    'buffalo_s':
-    'https://github.com/justld/dofaker/releases/download/v0.1/buffalo_s.zip',
-    'buffalo_sc':
-    'https://github.com/justld/dofaker/releases/download/v0.1/buffalo_sc.zip',
-    'inswapper':
-    'https://github.com/justld/dofaker/releases/download/v0.1/inswapper_128.onnx',
-    'gfpgan':
-    'https://github.com/justld/dofaker/releases/download/v0.1/GFPGANv1.3.onnx',
-    'bsrgan':
-    'https://github.com/justld/dofaker/releases/download/v0.1/bsrgan_4.onnx',
-    'openpose_body':
-    'https://github.com/justld/dofaker/releases/download/v0.1/openpose_body.onnx',
-    'pose_transfer':
-    'https://github.com/justld/dofaker/releases/download/v0.1/pose_transfer.onnx',
-}
-
-
-def get_model_url(model_name):
-    return WEIGHT_URLS[model_name]