Upload 2 files

ultralytics_solutions_modified/object_counter.py

@@ -50,8 +50,8 @@ class ObjectCounter:
         # Object counting Information
         self.in_counts = 0
         self.out_counts = 0
-        self.out_counts_prev = self.out_counts                            
-        self.in_counts_prev = self.in_counts                            
+        self.out_counts_prev = self.out_counts
+        self.in_counts_prev = self.in_counts
         self.counting_list = []
         self.count_txt_thickness = 0
         self.count_txt_color = (0, 0, 0)
@@ -264,7 +264,6 @@ class ObjectCounter:
                     if np.sign(
                         np.dot(obj_track_vec, self.counting_region_nvec)
                     ) < 0:
-                    # if (box[0] - prev_position[0]) * (self.counting_region.centroid.x - prev_position[0]) > 0:
                         self.out_counts += 1
                     else:
                         self.in_counts += 1
@@ -285,7 +284,6 @@ class ObjectCounter:
                         if np.sign(
                             np.dot(obj_track_vec, self.counting_region_nvec)
                         ) < 0:
-                        # if (box[0] - prev_position[0]) * (self.counting_region.centroid.x - prev_position[0]) > 0:
                             self.out_counts += 1
                         else:
                             self.in_counts += 1

ultralytics_solutions_modified/speed_estimation.py

@@ -139,7 +139,7 @@ class SpeedEstimator:
         self.annotator.box_label(box, box_label, bbox_color)
         cv2.polylines(
             self.im0, [self.trk_pts],
-            isClosed=False, color=(0, 255, 0), thickness=3
+            isClosed=False, color=(0, 255, 0), thickness=self.line_thickness
         )
         cv2.circle(
             self.im0, (int(track[-1][0]), int(track[-1][1])), 5,