Skip to content

Instantly share code, notes, and snippets.

@curiousily
Created February 17, 2020 21:14
Show Gist options
  • Save curiousily/ef5c4a7f5c2ce2666d17c423d5ccc4ba to your computer and use it in GitHub Desktop.
Save curiousily/ef5c4a7f5c2ce2666d17c423d5ccc4ba to your computer and use it in GitHub Desktop.
cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.85
predictor = DefaultPredictor(cfg)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment