import threading
|
import socket
|
import re
|
import time
|
import datetime
|
import os
|
import numpy as np
|
import pandas as pd
|
import faiss
|
import cv2
|
from flask import Flask
|
from flask import request
|
from flask import render_template
|
|
# live stream for channel 0, change to right url first
|
live_stream_0 = "http://192.168.77.60/flv?app=live&stream=lpr_0"
|
|
lpr_df = pd.DataFrame(columns=["time", "color", "text", "cropped_image", "whole_image"])
|
lpr_df['time'] = pd.to_datetime(lpr_df['time'])
|
|
'''
|
receive license plate recognize result from socket (sent from VideoPipe)
|
'''
|
def lpr_receiver():
|
udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
bind_ip = '192.168.77.68'
|
bind_port = 9999
|
udp_socket.bind((bind_ip, bind_port))
|
|
while True:
|
data, addr = udp_socket.recvfrom(1024 * 64)
|
s = data.decode()
|
print('############################################')
|
print(f'Received data: \n{s}')
|
print(f'From address: {addr}')
|
|
li = re.findall("\<\-\-.*?\-\-\>", s, re.DOTALL)
|
print(f"total {len(li)} records parsed! going to append to data frame...")
|
for l in li:
|
items = str(l).split('\n')
|
'''
|
items[0] <--
|
items[1] time
|
items[2] channel index, frame index
|
items[3] cropped image's path
|
items[4] whole image's path
|
items[5] license plate color
|
items[6] license plate text
|
items[7] -->
|
'''
|
assert(len(items) == 8)
|
# append items to dataframe
|
lpr_df.loc[len(lpr_df)] = [items[1].split('.')[0], items[5], items[6], items[3], items[4]]
|
print(f"now total {len(lpr_df)} records in lpr_df!")
|
print('############################################')
|
|
'''
|
create flask app
|
'''
|
app = Flask(__name__)
|
|
# home page
|
@app.route("/", methods=["GET", "POST"])
|
def home():
|
'''
|
filter properties
|
'''
|
select_channel = "all"
|
if 'select_channel' in request.form:
|
select_channel = request.form['select_channel']
|
|
# 1 hour by default
|
now = time.time()
|
past_1_hour = now - 3600
|
select_start_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(past_1_hour))
|
select_end_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(now))
|
|
if 'select_start_time' in request.form and request.form['select_start_time'] != '':
|
select_start_time = datetime.datetime.strptime(request.form['select_start_time'], '%Y-%m-%dT%H:%M:%S')
|
|
if 'select_end_time' in request.form and request.form['select_end_time'] != '':
|
select_end_time = datetime.datetime.strptime(request.form['select_end_time'], '%Y-%m-%dT%H:%M:%S')
|
|
sql = "color != ''"
|
if select_channel != 'all':
|
sql += f" & channel == '{select_channel}'"
|
sql += f" & time >= '{select_start_time}'"
|
sql += f" & time < '{select_end_time}'"
|
|
print(sql)
|
filter_df = lpr_df.query(sql)
|
print(filter_df)
|
plate_results = []
|
for index, row in filter_df.iterrows():
|
plate_results.append({'uid':index, \
|
'time':row['time'], \
|
'color':row['color'], \
|
'text':row['text'], \
|
'cropped_image': row['cropped_image'], \
|
'whole_image': row['whole_image']})
|
return render_template("home.html", \
|
plate_results=plate_results, \
|
total_records=len(filter_df), \
|
select_start_time = select_start_time, \
|
select_end_time = select_end_time, \
|
live_stream_0=live_stream_0)
|
|
if __name__ == "__main__":
|
# start plate receiving thread
|
receiver = threading.Thread(target = lpr_receiver)
|
receiver.start()
|
|
# run server
|
app.run(host="192.168.77.68", port=9696)
|
