Skip to content

M
maskrcnn
Commit c2619ed4 authored by CoinCheung's avatar CoinCheung Committed by Francisco Massa
Browse files
Options

readme and predictor.py (#233) 

* readme and predictor.py

* format modification and readme
parent d74fad1a
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
demo/predictor.py
View file @ c2619ed4
......@@ -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()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment