first commit

annotator/uniformer_base/__init__.py

@@ -22,12 +22,12 @@ checkpoint_file = "https://huggingface.co/Salesforce/UniControl/blob/main/annota
 
 class UniformerDetector:
     def __init__(self):
-        # modelpath = os.path.join(annotator_ckpts_path, "upernet_global_base.pth")
-        # if not os.path.exists(modelpath):
-        #     from basicsr.utils.download_util import load_file_from_url
-        #     load_file_from_url(checkpoint_file, model_dir=annotator_ckpts_path)
-#             raise ValueError("wrong ckpt path")
-        modelpath = checkpoint_file
+        modelpath = os.path.join(annotator_ckpts_path, "upernet_global_base.pth")
+        if not os.path.exists(modelpath):
+            from basicsr.utils.download_util import load_file_from_url
+            load_file_from_url(checkpoint_file, model_dir=annotator_ckpts_path)
+            raise ValueError("wrong ckpt path")
+        #modelpath = checkpoint_file
         config_file = os.path.join(os.path.dirname(annotator_ckpts_path), "uniformer_base", "exp", "upernet_global_base", "config.py")
         self.model = init_segmentor(config_file, modelpath).cuda()