Spaces:

Cyril666
/

ContourNet-ABI

Runtime error

Cyril666 commited on Jun 2, 2022

Commit

c6ed64e

1 Parent(s): 9ee5726

Update det_demo.py

Files changed (1) hide show

det_demo.py CHANGED Viewed

@@ -70,6 +70,7 @@ class DetDemo(object):
         self.min_image_size = min_image_size
         checkpointer = DetectronCheckpointer(cfg, self.model, save_dir=cfg.OUTPUT_DIR)
         _ = checkpointer.load(cfg.MODEL.WEIGHT)
         self.transforms = self.build_transform()

         self.min_image_size = min_image_size
         checkpointer = DetectronCheckpointer(cfg, self.model, save_dir=cfg.OUTPUT_DIR)
+        print(cfg.OUTPUT_DIR, cfg.MODEL.WEIGHT)
         _ = checkpointer.load(cfg.MODEL.WEIGHT)
         self.transforms = self.build_transform()