Spaces:
Sleeping
Sleeping
nn
commited on
Upload app.py
Browse files
app.py
CHANGED
|
@@ -179,7 +179,7 @@ def detect_objects(image, rail_segment):
|
|
| 179 |
|
| 180 |
fig = plot_boxes_and_segment(image, yolo_boxes, rail_segment, img_width, img_height, threshold)
|
| 181 |
|
| 182 |
-
|
| 183 |
for class_name, x, y, w, h in yolo_boxes:
|
| 184 |
if class_name != 'traffic light':
|
| 185 |
result = box_segment_relationship((0, x, y, w, h), rail_segment, img_width, img_height, threshold)
|
|
|
|
| 179 |
|
| 180 |
fig = plot_boxes_and_segment(image, yolo_boxes, rail_segment, img_width, img_height, threshold)
|
| 181 |
|
| 182 |
+
results = []
|
| 183 |
for class_name, x, y, w, h in yolo_boxes:
|
| 184 |
if class_name != 'traffic light':
|
| 185 |
result = box_segment_relationship((0, x, y, w, h), rail_segment, img_width, img_height, threshold)
|