Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def inference(image_url, image, min_score):
|
|
55 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
|
56 |
predictor = DefaultPredictor(cfg)
|
57 |
|
58 |
-
outputs = predictor(
|
59 |
|
60 |
v = Visualizer(image[:,:,::-1], my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
|
61 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
|
|
55 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
|
56 |
predictor = DefaultPredictor(cfg)
|
57 |
|
58 |
+
outputs = predictor(image)
|
59 |
|
60 |
v = Visualizer(image[:,:,::-1], my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
|
61 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|