Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -74,6 +74,10 @@ def inference(image_url, image, min_score, model_name):
|
|
74 |
|
75 |
v = Visualizer(im, models[model_id]["metadata"], scale=1.2)
|
76 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
|
|
|
|
|
|
|
|
77 |
|
78 |
return out.get_image()
|
79 |
|
|
|
74 |
|
75 |
v = Visualizer(im, models[model_id]["metadata"], scale=1.2)
|
76 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
77 |
+
|
78 |
+
# Convertir la imagen de salida de BGR a RGB
|
79 |
+
result_image = out.get_image() # Esto sigue estando en BGR
|
80 |
+
result_image_rgb = result_image[:, :, ::-1] # Convertir BGR a RGB
|
81 |
|
82 |
return out.get_image()
|
83 |
|