stalyn314 commited on
Commit
26ed2b6
·
verified ·
1 Parent(s): 9b12ea1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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