YarramsettiNaresh commited on
Commit
0883907
·
1 Parent(s): 2415f04

response change

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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
- object_count[class_name] = object_count.get(class_name, 0) + 1
 
 
 
 
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