Alessio Grancini commited on
Commit
a2a1782
·
verified ·
1 Parent(s): 5ef4344

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -259,8 +259,8 @@ def get_detection_data(image_data):
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({
@@ -271,8 +271,8 @@ def get_detection_data(image_data):
271
  },
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 = {
 
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({
 
271
  },
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 = {