Spaces:
Runtime error
Runtime error
Nadine Rueegg
commited on
Commit
β’
a7d1abf
1
Parent(s):
2a77bc9
small bug fix
Browse files
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')
|