Skip to content

Instantly share code, notes, and snippets.

@emadehsan

emadehsan/create-model.py

Created January 29, 2020 15:44
Show Gist options
  • Save emadehsan/c3f719d26c34c96918cdcfcfe375354f to your computer and use it in GitHub Desktop.
Save emadehsan/c3f719d26c34c96918cdcfcfe375354f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
create-model.py
# Create model object in inference mode.
model = modellib.MaskRCNN(mode="inference", model_dir=MODEL_DIR, config=config)
# Load weights trained on MS-COCO
model.load_weights(COCO_MODEL_PATH, by_name=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment