Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
@@ -50,14 +50,14 @@ def inference(image_url, image, min_score):
|
|
50 |
im = cv2.imdecode(im, cv2.IMREAD_COLOR)
|
51 |
else:
|
52 |
# Model expect BGR!
|
53 |
-
im = image[:,:,::-1]
|
54 |
|
55 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
|
56 |
predictor = DefaultPredictor(cfg)
|
57 |
|
58 |
-
outputs = predictor(
|
59 |
|
60 |
-
v = Visualizer(im, my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
|
61 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
62 |
|
63 |
return out.get_image()
|
|
|
50 |
im = cv2.imdecode(im, cv2.IMREAD_COLOR)
|
51 |
else:
|
52 |
# Model expect BGR!
|
53 |
+
im = image#[:,:,::-1]
|
54 |
|
55 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
|
56 |
predictor = DefaultPredictor(cfg)
|
57 |
|
58 |
+
outputs = predictor(im)
|
59 |
|
60 |
+
v = Visualizer(im[:,:,::-1], my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
|
61 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
62 |
|
63 |
return out.get_image()
|