Spaces:
Sleeping
Sleeping
Commit
·
0883907
1
Parent(s):
2415f04
response change
Browse files
app.py
CHANGED
@@ -31,7 +31,11 @@ def ultralytics_predict(model, frame):
|
|
31 |
# Update object count
|
32 |
class_id = int(detection.cls[0])
|
33 |
class_name = model.names[class_id]
|
34 |
-
|
|
|
|
|
|
|
|
|
35 |
|
36 |
return object_count # Return the count of detected objects
|
37 |
|
|
|
31 |
# Update object count
|
32 |
class_id = int(detection.cls[0])
|
33 |
class_name = model.names[class_id]
|
34 |
+
if class_name not in object_count:
|
35 |
+
object_count[class_name] = dict(count=0, objects=[])
|
36 |
+
object_mapp = object_count[class_name]
|
37 |
+
object_mapp["count"] = object_mapp.get("count", 0) + 1
|
38 |
+
object_mapp["objects"].append(dict(conf=conf, pos=pos, text=text, color=color))
|
39 |
|
40 |
return object_count # Return the count of detected objects
|
41 |
|