Update app.py

app.py

@@ -37,15 +37,29 @@ full_plate_model, character_model = load_models()
 # Function to detect and crop license plates
 def detect_and_crop_license_plate(image):
     img_bgr = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
-    results = full_plate_model(img_bgr)
-    
+    results = full_plate_model(img_bgr, conf=0.5)  # Use a higher confidence threshold
+
     cropped_images = []
+    h, w, _ = img_bgr.shape
     for result in results:
         if hasattr(result, 'boxes') and result.boxes is not None:
             for box in result.boxes.xyxy:
                 x1, y1, x2, y2 = map(int, box)
-                cropped_image = img_bgr[y1:y2, x1:x2]
-                cropped_images.append(cropped_image)
+
+                # Add padding to the bounding box
+                padding = 10
+                x1 = max(0, x1 - padding)
+                y1 = max(0, y1 - padding)
+                x2 = min(w, x2 + padding)
+                y2 = min(h, y2 + padding)
+
+                # Avoid extremely small detections
+                if (x2 - x1) > 20 and (y2 - y1) > 20:
+                    cropped_image = img_bgr[y1:y2, x1:x2]
+
+                    # Resize to a consistent size
+                    cropped_image = cv2.resize(cropped_image, (224, 224), interpolation=cv2.INTER_LINEAR)
+                    cropped_images.append(cropped_image)
 
     return cropped_images, img_bgr