Skip to content

Instantly share code, notes, and snippets.

@THEFASHIONGEEK

THEFASHIONGEEK/rfbnet_Inference.py

Created April 9, 2020 17:00
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 THEFASHIONGEEK/cd4408693d34ce3013b64b0c6ed89621 to your computer and use it in GitHub Desktop.
Save THEFASHIONGEEK/cd4408693d34ce3013b64b0c6ed89621 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
rfbnet_Inference.py
from infer_detector import Infer
gtf = Infer()
gtf.Model(model_name="mobilenet", weights="weights/Final_RFB_mobile_COCO.pth", use_gpu=True)
gtf.Image_Params(class_file, input_size=300)
gtf.Setup()
output = gtf.Predict(img_path, thresh=0.195, font_size=1, line_size=3)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment