Update app.py

app.py

@@ -211,13 +211,15 @@ def process_video(confidence_threshold=0.5, selected_classes=None, stream_url=No
             errors.append("Error: Could not read frame from the stream.")
             break
 
+        # Get the original frame dimensions
+        original_height, original_width = frame.shape[:2]
+
         # Resize the frame for processing
-        height, width = frame.shape[:2]
-        new_width = int(width * resolution_scale)
-        new_height = int(height * resolution_scale)
+        new_width = int(original_width * resolution_scale)
+        new_height = int(original_height * resolution_scale)
         resized_frame = cv2.resize(frame, (new_width, new_height))
 
-        # Perform object tracking with confidence threshold
+        # Perform object tracking with confidence threshold on the resized frame
         results = model.track(resized_frame, persist=True, conf=confidence_threshold)
 
         if results[0].boxes.id is not None:
@@ -229,7 +231,7 @@ def process_video(confidence_threshold=0.5, selected_classes=None, stream_url=No
             for box, cls, t_id, conf in zip(boxes, clss, track_ids, confs):
                 if conf >= confidence_threshold and model.names[cls] in selected_classes:
                     # Scale the bounding box back to the original resolution
-                    box = box * (width / new_width)
+                    box = box * (original_width / new_width)
                     # Check if the object crosses the line
                     if is_object_crossing_line(box, line_params) and t_id not in crossed_objects:
                         crossed_objects[t_id] = True
@@ -241,7 +243,7 @@ def process_video(confidence_threshold=0.5, selected_classes=None, stream_url=No
         # Visualize the results with bounding boxes, masks, and IDs on the original frame
         annotated_frame = results[0].plot(img=frame)
 
-        # Draw the angled line on the frame
+        # Draw the angled line on the original frame
         draw_angled_line(annotated_frame, line_params, color=(0, 255, 0), thickness=2)
 
         # Display the count on the frame with a modern look