Remove labels for the unsupervised model

.pre-commit-config.yaml

@@ -1,3 +1,4 @@
+exclude: patch
 repos:
 - repo: https://github.com/pre-commit/pre-commit-hooks
   rev: v4.2.0

Dockerfile

@@ -50,6 +50,7 @@ RUN pip install --no-cache-dir -U \
 RUN pip install --no-cache-dir -U gradio==3.16.2
 
 COPY --chown=1000 . ${HOME}/app
+RUN cd CutLER && git apply ../patch
 ENV PYTHONPATH=${HOME}/app \
 	PYTHONUNBUFFERED=1 \
 	GRADIO_ALLOW_FLAGGING=never \

model.py

@@ -131,6 +131,8 @@ class Model:
             weight_path.as_posix(), 'MODEL.DEVICE',
             'cuda:0' if torch.cuda.is_available() else 'cpu'
         ]
+        if model_name in UNSUPERVISED_MODELS:
+            arg_list += ['DATASETS.TEST', '()']
         args = get_parser().parse_args(arg_list)
         cfg = setup_cfg(args)
         return VisualizationDemo(cfg)

patch

@@ -0,0 +1,13 @@
+diff --git a/cutler/demo/predictor.py b/cutler/demo/predictor.py
+index 0920159..6fc8053 100644
+--- a/cutler/demo/predictor.py
++++ b/cutler/demo/predictor.py
+@@ -62,6 +62,8 @@ class VisualizationDemo(object):
+                 )
+             if "instances" in predictions:
+                 instances = predictions["instances"].to(self.cpu_device)
++                if self.metadata.name == '__unused':
++                    instances.remove('pred_classes')
+                 vis_output = visualizer.draw_instance_predictions(predictions=instances)
+ 
+         return predictions, vis_output