karthikmn commited on
Commit
cd1ebf9
·
verified ·
1 Parent(s): 9ef2b34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -4,6 +4,7 @@ import cv2
4
  import pandas as pd
5
  import smtplib
6
  from datetime import datetime, timedelta
 
7
 
8
  # Load pre-trained model
9
  model_name = "hustvl/yolos-small"
@@ -27,10 +28,14 @@ def detect_number_plates(image):
27
 
28
  return number_plates
29
 
30
- def save_to_excel(number_plates):
 
 
 
 
31
  # Create a pandas DataFrame
32
  df = pd.DataFrame({
33
- "Date": [datetime.now().strftime("%Y-%m-%d %H:%M:%S") for _ in range(len(number_plates))],
34
  "Number Plate": [f"{x1}, {y1}, {x2}, {y2}" for x1, y1, x2, y2 in number_plates]
35
  })
36
 
@@ -41,6 +46,7 @@ def detect_number_plates(image):
41
  combined_df.to_excel("number_plates.xlsx", index=False)
42
  except FileNotFoundError:
43
  df.to_excel("number_plates.xlsx", index=False)
 
44
  def monitor_vehicles(number_plates):
45
  # Load registered number plates from a file or database
46
  registered_plates = pd.read_csv("registered_plates.csv")["Number Plate"].tolist()
 
4
  import pandas as pd
5
  import smtplib
6
  from datetime import datetime, timedelta
7
+ import datetime
8
 
9
  # Load pre-trained model
10
  model_name = "hustvl/yolos-small"
 
28
 
29
  return number_plates
30
 
31
+ def detect_number_plates(image):
32
+ # Your code here
33
+ pass
34
+
35
+ def save_to_excel(number_plates):
36
  # Create a pandas DataFrame
37
  df = pd.DataFrame({
38
+ "Date": [datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") for _ in range(len(number_plates))],
39
  "Number Plate": [f"{x1}, {y1}, {x2}, {y2}" for x1, y1, x2, y2 in number_plates]
40
  })
41
 
 
46
  combined_df.to_excel("number_plates.xlsx", index=False)
47
  except FileNotFoundError:
48
  df.to_excel("number_plates.xlsx", index=False)
49
+
50
  def monitor_vehicles(number_plates):
51
  # Load registered number plates from a file or database
52
  registered_plates = pd.read_csv("registered_plates.csv")["Number Plate"].tolist()