local test

MODEL/local1.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fc1399cf65504852e03586a1d665f0613c33baff7228d06c1c061a30961174ab
+size 3709941017

MODEL/local2.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b696e97ef9e82ae5e080f51ea0e50134d78e2cfd336b8d75590ee9b53d9f2f51
+size 3709919257

app.py

@@ -49,6 +49,34 @@ fiber_demo = GLIPDemo(
     show_mask_heatmaps=False
 )
 
+config_file = "configs/pretrain_new/desco_glip.yaml"
+weight_file = "MODEL/local1.pth"
+from copy import deepcopy
+cfg = deepcopy(cfg)
+cfg.merge_from_file(config_file)
+cfg.merge_from_list(["MODEL.WEIGHT", weight_file])
+cfg.merge_from_list(["MODEL.DEVICE", "cuda"])
+local_demo1 = GLIPDemo(
+    cfg,
+    min_image_size=800,
+    confidence_threshold=0.7,
+    show_mask_heatmaps=False
+)
+
+config_file = "configs/pretrain_new/desco_glip.yaml"
+weight_file = "MODEL/local2.pth"
+from copy import deepcopy
+cfg = deepcopy(cfg)
+cfg.merge_from_file(config_file)
+cfg.merge_from_list(["MODEL.WEIGHT", weight_file])
+cfg.merge_from_list(["MODEL.DEVICE", "cuda"])
+local_demo2 = GLIPDemo(
+    cfg,
+    min_image_size=800,
+    confidence_threshold=0.7,
+    show_mask_heatmaps=False
+)
+
 athetics_params = {
     "skip_name": False, # whether we overlay the phrase over the box
     "override_color": (0, 90, 190),
@@ -67,7 +95,9 @@ def predict(image, text, ground_tokens=""):
     ground_tokens = None if ground_tokens.strip() == "" else ground_tokens.strip().split(";")
     result, _ = glip_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, ground_tokens, **athetics_params)
     fiber_result, _ = fiber_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, ground_tokens, **athetics_params)
-    return result[:, :, [2, 1, 0]], fiber_result[:, :, [2, 1, 0]]
+    local_result1, _ = local_demo1.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, ground_tokens, **athetics_params)
+    local_result2, _ = local_demo2.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, ground_tokens, **athetics_params)
+    return result[:, :, [2, 1, 0]], fiber_result[:, :, [2, 1, 0]], local_result1, local_result2
 
 
 image = gr.inputs.Image()
@@ -87,6 +117,14 @@ gr.Interface(
             type="pil",
             label="DesCo-FIBER"
         ),
+        gr.outputs.Image(
+            type="pil",
+            label="local1"
+        ),
+        gr.outputs.Image(
+            type="pil",
+            label="local2"
+        ),
     ],
     examples=[
         ["./1.jpg", "A clown making a balloon animal for a pretty lady.", "clown"],