Nadine Rueegg commited on
Commit
a7d1abf
β€’
1 Parent(s): 2a77bc9

small bug fix

Browse files
Files changed (1) hide show
  1. gradio_demo/barc_demo_v3.py +3 -0
gradio_demo/barc_demo_v3.py CHANGED
@@ -107,6 +107,9 @@ def detect_object(model, img_path_or_img, confidence=0.5, rect_th=2, text_size=0
107
 
108
 
109
  def run_bbox_inference(input_image):
 
 
 
110
  model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
111
  model.eval()
112
  out_path = os.path.join(cfg.paths.ROOT_OUT_PATH, 'gradio_examples', 'test2.png')
 
107
 
108
 
109
  def run_bbox_inference(input_image):
110
+ # load configs
111
+ cfg = get_cfg_global_updated()
112
+
113
  model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
114
  model.eval()
115
  out_path = os.path.join(cfg.paths.ROOT_OUT_PATH, 'gradio_examples', 'test2.png')