fiber

app.py

@@ -65,6 +65,8 @@ glip_demo = GLIPDemo(
 
 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"])
@@ -80,7 +82,6 @@ def predict(image, text, specified_tokens=""):
         specified_tokens = None
     else:
         specified_tokens = specified_tokens.strip().split(";")
-    from copy import deepcopy
     result, _ = glip_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, specified_tokens)
     fiber_result, _ = fiber_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, specified_tokens)
     return result[:, :, [2, 1, 0]], fiber_result[:, :, [2, 1, 0]]