Alessio Grancini commited on
Update app.py
Browse files
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 |
-
|
| 263 |
-
|
| 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(
|
| 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 = {
|