Update app.py
Browse files
app.py
CHANGED
@@ -37,8 +37,8 @@ def detect_objects_with_sahi(image):
|
|
37 |
|
38 |
# Count objects by class
|
39 |
class_counts = {}
|
40 |
-
for
|
41 |
-
class_name =
|
42 |
class_counts[class_name] = class_counts.get(class_name, 0) + 1
|
43 |
|
44 |
# Create summary text
|
|
|
37 |
|
38 |
# Count objects by class
|
39 |
class_counts = {}
|
40 |
+
for i in range(len(sliced_detections.class_id)): # Iterate over the detections
|
41 |
+
class_name = sliced_detections.class_id[i]
|
42 |
class_counts[class_name] = class_counts.get(class_name, 0) + 1
|
43 |
|
44 |
# Create summary text
|