Pablo Barrera Yaksic hai 2 semanas
achega
bc6ebe7b58
Modificáronse 6 ficheiros con 127 adicións e 0 borrados
  1. 35 0
      main.py
  2. BIN=BIN
      plates/1.jpg
  3. BIN=BIN
      plates/2.jpg
  4. BIN=BIN
      plates/3.jpg
  5. BIN=BIN
      plates/4.png
  6. 92 0
      requirements.txt

+ 35 - 0
main.py

@@ -0,0 +1,35 @@
+import os
+import shutil
+import re
+
+from yolov5 import detect
+import easyocr
+
+def main():
+    directory = "plates"
+    detect.run(
+        source=directory,
+        save_crop=True,
+        weights="yolo-licence-plate.pt",
+        conf_thres=0.25,
+        iou_thres=0.45,
+        agnostic_nms=False,
+        max_det=1,
+        project="predictions",
+        exist_ok=True
+    )
+
+    reader = easyocr.Reader(["es"])
+    crops_directory = "predictions/exp/crops/license_plate"
+    for image in os.listdir(crops_directory):
+        filepath = os.path.join(crops_directory, image)
+        if os.path.isfile(filepath) and image.endswith((".jpg", ".jpeg", ".png")):
+            image_path = os.path.join(crops_directory, image)
+            print(f"Processing {image_path}...")
+            result = reader.readtext(image_path, detail=0)
+            clean_plate = re.sub(r"[^A-Z0-9]", "", result[0].upper())
+            print(clean_plate)
+
+    shutil.rmtree("predictions/exp")
+if __name__ == "__main__":
+    main()

BIN=BIN
plates/1.jpg


BIN=BIN
plates/2.jpg


BIN=BIN
plates/3.jpg


BIN=BIN
plates/4.png


+ 92 - 0
requirements.txt

@@ -0,0 +1,92 @@
+absl-py==2.2.2
+boto3==1.38.2
+botocore==1.38.2
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+contourpy==1.3.2
+cycler==0.12.1
+easyocr==1.7.2
+filelock==3.18.0
+filetype==1.2.0
+fire==0.7.0
+fonttools==4.57.0
+fsspec==2025.3.2
+gitdb==4.0.12
+GitPython==3.1.44
+grpcio==1.72.0
+huggingface-hub==0.24.7
+idna==3.7
+imageio==2.37.0
+Jinja2==3.1.6
+jmespath==1.0.1
+kiwisolver==1.4.8
+lazy_loader==0.4
+Markdown==3.8
+MarkupSafe==3.0.2
+matplotlib==3.10.1
+mpmath==1.3.0
+networkx==3.4.2
+ninja==1.11.1.4
+numpy==2.1.1
+nvidia-cublas-cu12==12.6.4.1
+nvidia-cuda-cupti-cu12==12.6.80
+nvidia-cuda-nvrtc-cu12==12.6.77
+nvidia-cuda-runtime-cu12==12.6.77
+nvidia-cudnn-cu12==9.5.1.17
+nvidia-cufft-cu12==11.3.0.4
+nvidia-cufile-cu12==1.11.1.6
+nvidia-curand-cu12==10.3.7.77
+nvidia-cusolver-cu12==11.7.1.2
+nvidia-cusparse-cu12==12.5.4.2
+nvidia-cusparselt-cu12==0.6.3
+nvidia-nccl-cu12==2.26.2
+nvidia-nvjitlink-cu12==12.6.85
+nvidia-nvtx-cu12==12.6.77
+opencv-python==4.10.0.84
+opencv-python-headless==4.10.0.84
+packaging==25.0
+pandas==2.2.3
+pillow==11.2.1
+pillow_heif==0.22.0
+protobuf==6.30.2
+psutil==7.0.0
+py-cpuinfo==9.0.0
+pybboxes==0.1.6
+pyclipper==1.3.0.post6
+pyparsing==3.2.3
+python-bidi==0.6.6
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.0
+pytz==2025.2
+PyYAML==6.0.2
+requests==2.32.3
+requests-toolbelt==1.0.0
+roboflow==1.1.61
+s3transfer==0.12.0
+sahi==0.11.22
+scikit-image==0.25.2
+scipy==1.15.2
+seaborn==0.13.2
+setuptools==79.0.1
+shapely==2.1.0
+six==1.17.0
+smmap==5.0.2
+sympy==1.13.3
+tensorboard==2.19.0
+tensorboard-data-server==0.7.2
+termcolor==3.0.1
+terminaltables==3.1.10
+thop==0.1.1.post2209072238
+tifffile==2025.3.30
+torch==2.7.0
+torchvision==0.22.0
+tqdm==4.67.1
+triton==3.3.0
+typing_extensions==4.13.2
+tzdata==2025.2
+ultralytics==8.3.116
+ultralytics-thop==2.0.14
+urllib3==2.4.0
+Werkzeug==3.1.3
+yolov5==7.0.14