local test

app.py

@@ -35,14 +35,14 @@ glip_demo = GLIPDemo(
     show_mask_heatmaps=False
 )
 
-config_file = "configs/pretrain_new/desco_fiber.yaml"
-weight_file = "MODEL/desco_fiber_base.pth"
+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"])
-fiber_demo = GLIPDemo(
+local_demo1 = GLIPDemo(
     cfg,
     min_image_size=800,
     confidence_threshold=0.7,
@@ -50,27 +50,27 @@ fiber_demo = GLIPDemo(
 )
 
 config_file = "configs/pretrain_new/desco_glip.yaml"
-weight_file = "MODEL/local1.pth"
+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_demo1 = GLIPDemo(
+local_demo2 = 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"
+config_file = "configs/pretrain_new/desco_fiber.yaml"
+weight_file = "MODEL/desco_fiber_base.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(
+fiber_demo = GLIPDemo(
     cfg,
     min_image_size=800,
     confidence_threshold=0.7,
@@ -97,7 +97,7 @@ def predict(image, text, ground_tokens=""):
     fiber_result, _ = fiber_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, ground_tokens, **athetics_params)
     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
+    return result[:, :, [2, 1, 0]], fiber_result[:, :, [2, 1, 0]], local_result1[:, :, [2, 1, 0]], local_result2[:, :, [2, 1, 0]]
 
 
 image = gr.inputs.Image()