Update model_1.py

model_1.py

@@ -48,10 +48,10 @@ color_mapping = {
     'Empty class': (0, 255, 0)          # Green for empty
 }
 
-def perform_detection(image):
+def perform_detection(image, target_size=(640, 640)):
     imH, imW, _ = image.shape
     image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-    image_resized = cv2.resize(image_rgb, (width, height))
+    image_resized = cv2.resize(image_rgb, target_size)
     input_data = np.expand_dims(image_resized, axis=0)
 
     if floating_model:
@@ -86,8 +86,10 @@ def perform_detection(image):
     return image
 
 def detect_image(input_image):
+    # Resize the image to 640x640 for faster processing
+    target_size = (640, 640)
     image = np.array(input_image)
-    result_image = perform_detection(image)
+    result_image = perform_detection(image, target_size)
     return Image.fromarray(result_image)
 
 def detect_video(input_video):
@@ -99,7 +101,7 @@ def detect_video(input_video):
         if not ret:
             break
 
-        result_frame = perform_detection(frame)
+        result_frame = perform_detection(frame, (640, 640))
         frames.append(result_frame)
 
     cap.release()