Skip to content

Instantly share code, notes, and snippets.

@Tony607

Tony607/predictor.py

Created October 14, 2019 05:35
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Tony607/b1b6222f3ddfc5b6b27c8ef4048f4b71 to your computer and use it in GitHub Desktop.
Save Tony607/b1b6222f3ddfc5b6b27c8ef4048f4b71 to your computer and use it in GitHub Desktop.
Download ZIP
How to train Detectron2 with Custom COCO Datasets | DLology
Raw
predictor.py
cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5 # set the testing threshold for this model
cfg.DATASETS.TEST = ("fruits_nuts", )
predictor = DefaultPredictor(cfg)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment