first commit

annotator/midas/api.py

@@ -89,10 +89,10 @@ def load_model(model_type):
     # load network
     model_path = ISL_PATHS[model_type]
     if model_type == "dpt_large":  # DPT-Large
-        # if not os.path.exists(model_path):
-        #     from basicsr.utils.download_util import load_file_from_url
-        #     load_file_from_url(remote_model_path, model_dir=annotator_ckpts_path)
-        model_path = remote_model_path
+        if not os.path.exists(model_path):
+            from basicsr.utils.download_util import load_file_from_url
+            load_file_from_url(remote_model_path, model_dir=annotator_ckpts_path)
+        #model_path = remote_model_path
         model = DPTDepthModel(
             path=model_path,
             backbone="vitl16_384",

requirements.txt

@@ -21,4 +21,5 @@ transformers==4.26.1
 xformers==0.0.16
 yapf==0.32.0
 cvlib==0.2.7
-tensorflow-cpu
+tensorflow-cpu
+basicsr==1.4.2