Add post request

johnng0805 · May 31, 2022 · d75d688 · d75d688
1 parent c593226
commit d75d688
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -254,3 +254,6 @@ com_crashlytics_export_strings.xml
 crashlytics.properties
 crashlytics-build.properties
 fabric.properties
+
+# videos folder
+videos/
diff --git a/detect.py b/detect.py
@@ -24,6 +24,7 @@
                                          yolov5s_edgetpu.tflite     # TensorFlow Edge TPU
 """
 
+import requests
 from utils.openalpr import Alpr
 from utils.torch_utils import select_device, time_sync
 from utils.plots import Annotator, colors, save_one_box
@@ -40,6 +41,9 @@
 import torch
 import torch.backends.cudnn as cudnn
 
+from dotenv import load_dotenv
+load_dotenv()
+
 FILE = Path(__file__).resolve()
 ROOT = FILE.parents[0]  # YOLOv5 root directory
 if str(ROOT) not in sys.path:
@@ -104,6 +108,13 @@ def run(
         alpr.set_top_n(1)
         alpr.set_detect_region(False)
 
+    # Declare API Endpoint
+    API_ENDPOINT = os.environ['API_ENDPOINT']
+    print('Application Backend: ' + API_ENDPOINT)
+
+    # Previous plate value
+    prev_plate = ''
+
     # Dataloader
     if webcam:
         view_img = check_imshow()
@@ -182,13 +193,25 @@ def run(
                     #     save_one_box(xyxy, imc, file=save_dir / 'crops' / names[c] / f'{p.stem}.jpg', BGR=True)
                     c = int(cls)
                     if c == 0:
-                        label = None
                         imCrop = crop(xyxy, imc)
                         plate_num = alpr.recognize_ndarray(imCrop)
 
                         if (plate_num['results'] != []):
                             plate = plate_num['results'][0]['plate']
-                            print(plate)
+
+                            if plate and plate != prev_plate:
+                                data = {'plate_num': plate}
+
+                                try:
+                                    r = requests.post(API_ENDPOINT, json=data, timeout=1.5)
+                                    prev_plate = plate
+                                    print(r.text)
+                                except requests.Timeout:
+                                    print('Connection timeout')
+                                    pass
+                                except requests.ConnectionError:
+                                    print('Connection error')
+                                    pass
 
             # Stream results
             im0 = annotator.result()

diff --git a/utils/image_crop.py b/utils/image_crop.py
@@ -19,6 +19,8 @@ def crop(xyxy, im, square=False, gain=1.02, pad=10, BGR=False):
 
     crop = im[int(xyxy[0, 1]):int(xyxy[0, 3]), int(xyxy[0, 0]):int(xyxy[0, 2]), ::(1 if BGR else -1)]
 
+    # crop = cv2.resize(crop, None, fx=2, fy=2, interpolation=cv2.INTER_CUBIC)
+
     gray = cv2.cvtColor(crop, cv2.COLOR_RGB2GRAY)
 
     medBlur = cv2.medianBlur(gray, 3)