readme and predictor.py (#233)

* readme and predictor.py * format modification and readme

readme and predictor.py (#233)
* readme and predictor.py * format modification and readme
c2619ed4 · CoinCheung · Francisco Massa · d74fad1a · c2619ed4
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

demo/predictor.py demo/predictor.py +2 -1

未找到文件。
--- a/demo/predictor.py
+++ b/demo/predictor.py
@@ -110,7 +110,8 @@ class COCODemo(object):
        self.model.to(self.device)
        self.min_image_size = min_image_size

-        checkpointer = DetectronCheckpointer(cfg, self.model)
+        save_dir = cfg.OUTPUT_DIR
+        checkpointer = DetectronCheckpointer(cfg, self.model, save_dir=save_dir)
        _ = checkpointer.load(cfg.MODEL.WEIGHT)

        self.transforms = self.build_transform()