Update detection/base.py

detection/base.py

@@ -53,18 +53,18 @@ class ObjectDetection:
 
     def _gr_detect(self, image: ImageTyping, model_name: str,
                    iou_threshold: float = 0.7, score_threshold: float = 0.25) \
-            -> gr.AnnotatedImage:
+            -> Tuple[gr.AnnotatedImage, List[Tuple[Tuple[float, float, float, float], str, float]]]:
         labels = self.get_labels(model_name=model_name)
         _colors = list(map(str, rnd_colors(len(labels))))
         _color_map = dict(zip(labels, _colors))
+        detections = self.detect(image, model_name, iou_threshold, score_threshold)
         return gr.AnnotatedImage(
             value=(image, [
-                (_bbox_fix(bbox), label) for bbox, label, _ in
-                self.detect(image, model_name, iou_threshold, score_threshold)
+                (_bbox_fix(bbox), label) for bbox, label, _ in detections
             ]),
             color_map=_color_map,
             label='Labeled',
-        )
+        ), detections
 
     def make_ui(self):
         with gr.Row():
@@ -82,6 +82,7 @@ class ObjectDetection:
 
             with gr.Column():
                 gr_output_image = gr.AnnotatedImage(label="Labeled")
+                gr_detections = gr.JSON(label="Detections")
 
             gr_submit.click(
                 self._gr_detect,
@@ -91,7 +92,7 @@ class ObjectDetection:
                     gr_iou_threshold,
                     gr_score_threshold,
                 ],
-                outputs=[gr_output_image],
+                outputs=[gr_output_image, gr_detections],
             )