Alessio Grancini commited on
Commit
2ce0c64
·
verified ·
1 Parent(s): 8d79b8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -258,9 +258,10 @@ def get_detection_data(image_data):
258
  cls_id, cls_name, center, mask, _ = obj
259
  x1, y1, x2, y2 = get_bbox_from_mask(mask)
260
  depth_value = depth_at_center(depthmap, [x1, y1, x2, y2])
 
261
  # Get color in RGB format (OpenCV uses BGR)
262
- color_bgr = objects_data[idx][4] # cls_clr from the objects_data
263
- color_rgb = (color_bgr[2], color_bgr[1], color_bgr[0]) # Convert BGR to RGB
264
 
265
 
266
  detections.append({
@@ -272,7 +273,7 @@ def get_detection_data(image_data):
272
  "position_3d": get_3d_position(center, depth_value, get_camera_matrix(depth_estimator)),
273
  "distance": depth_value,
274
  "color": color_rgb,
275
- "confidence": float(objects_data[idx][5]) # Add confidence score
276
  })
277
 
278
  response = {
 
258
  cls_id, cls_name, center, mask, _ = obj
259
  x1, y1, x2, y2 = get_bbox_from_mask(mask)
260
  depth_value = depth_at_center(depthmap, [x1, y1, x2, y2])
261
+
262
  # Get color in RGB format (OpenCV uses BGR)
263
+ color_rgb = (int(color_bgr[2]), int(color_bgr[1]), int(color_bgr[0])) # Convert BGR to RGB
264
+
265
 
266
 
267
  detections.append({
 
273
  "position_3d": get_3d_position(center, depth_value, get_camera_matrix(depth_estimator)),
274
  "distance": depth_value,
275
  "color": color_rgb,
276
+ "confidence": float(confidence)
277
  })
278
 
279
  response = {