Skip to content

A
aart_ppe_detection
Commit 494a31c4 authored by Sikhin VC's avatar Sikhin VC
Browse files
Options

added api recording feature

parent ae9b6670
Show whitespace changes
Inline Side-by-side
Showing with 16 additions and 16 deletions
scripts/ppe.py
View file @ 494a31c4
......@@ -454,7 +454,7 @@ class Ppe(ModelWrapper):
# present_event_id = 'SELECT "event_id" FROM aarti_violation_event_table'
cur = self.conn.cursor()
frame_count = self.frame_id
file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".avi"
file_path = "/home/administrator/aarti_violation_videos/" + str(unique_id) + ".webm"
cur.execute(sql, [frame_count, 4, str(unique_id), datetime.datetime.now(), file_path, v])
self.conn.commit()
cur.close()
......@@ -463,7 +463,7 @@ class Ppe(ModelWrapper):
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write(rec_inf[2] + ".webm")
f.write("\n")
del self.active_rec[object_id]
......@@ -556,7 +556,7 @@ class Ppe(ModelWrapper):
# present_event_id = 'SELECT "event_id" FROM aarti_violation_event_table'
cur = self.conn.cursor()
frame_count = self.frame_id
file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".avi"
file_path = "/home/administrator/aarti_violation_videos/" + str(unique_id) + ".webm"
cur.execute(sql,
[frame_count, 4, str(unique_id), datetime.datetime.now(), file_path,
v])
......@@ -567,7 +567,7 @@ class Ppe(ModelWrapper):
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write(rec_inf[2] + ".webm")
f.write("\n")
del self.active_rec[object_id]
......@@ -597,7 +597,7 @@ class Ppe(ModelWrapper):
cur = self.conn.cursor()
# cur.execute("ROLLBACK")
frame_count = self.frame_id
file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".avi"
file_path = "/home/administrator/aarti_violation_videos/" + str(unique_id) + ".webm"
cur.execute(sql, [frame_count, 4, str(unique_id), datetime.datetime.now(), file_path, v])
self.conn.commit()
cur.close()
......@@ -607,7 +607,7 @@ class Ppe(ModelWrapper):
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write(rec_inf[2] + ".webm")
f.write("\n")
del self.active_rec[object_id]
......@@ -644,7 +644,7 @@ class Ppe(ModelWrapper):
# unique_id = uuid.uuid4()
unique_id = str(uuid.uuid4()) + str(time.time())
# with open("recorded_video_list.txt", "a") as f:
# f.write(unique_id + ".avi")
# f.write(unique_id + ".webm")
# f.write("\n")
# result = self.video_write(frame, unique_id)
requests.post('http://localhost:2328/items/', json={
......@@ -675,7 +675,7 @@ class Ppe(ModelWrapper):
# cur = self.conn.cursor()
# # cur.execute("ROLLBACK")
# frame_count = self.frame_id
# file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".avi"
# file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".webm"
# cur.execute(sql, [frame_count, 4, str(unique_id), datetime.datetime.now(), file_path, v])
# self.conn.commit()
# cur.close()
......@@ -700,7 +700,7 @@ class Ppe(ModelWrapper):
cur = self.conn.cursor()
# cur.execute("ROLLBACK")
frame_count = self.frame_id
file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".avi"
file_path = "/home/administrator/aarti_violation_videos/" + str(unique_id) + ".webm"
cur.execute(sql,
[frame_count, 4, str(unique_id), datetime.datetime.now(), file_path,
v])
......@@ -711,7 +711,7 @@ class Ppe(ModelWrapper):
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write(rec_inf[2] + ".webm")
f.write("\n")
del self.active_rec[object_id]
......@@ -739,7 +739,7 @@ class Ppe(ModelWrapper):
cur = self.conn.cursor()
# cur.execute("ROLLBACK")
frame_count = self.frame_id
file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".avi"
file_path = "/home/administrator/aarti_violation_videos/" + str(unique_id) + ".webm"
cur.execute(sql, [frame_count, 4, str(unique_id), datetime.datetime.now(), file_path, v])
self.conn.commit()
cur.close()
......@@ -749,7 +749,7 @@ class Ppe(ModelWrapper):
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write(rec_inf[2] + ".webm")
f.write("\n")
del self.active_rec[object_id]
......@@ -784,7 +784,7 @@ class Ppe(ModelWrapper):
# unique_id = uuid.uuid4()
unique_id = str(uuid.uuid4()) + str(time.time())
# with open("recorded_video_list.txt", "a") as f:
# f.write(unique_id + ".avi")
# f.write(unique_id + ".webm")
# f.write("\n")
# result = self.video_write(frame, unique_id)
requests.post('http://localhost:2328/items/', json={
......@@ -815,7 +815,7 @@ class Ppe(ModelWrapper):
# cur = self.conn.cursor()
# # cur.execute("ROLLBACK")
# frame_count = self.frame_id
# file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".avi"
# file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".webm"
# cur.execute(sql, [frame_count, 4, str(unique_id), datetime.datetime.now(), file_path, v])
# self.conn.commit()
# cur.close()
......@@ -844,7 +844,7 @@ class Ppe(ModelWrapper):
cur = self.conn.cursor()
# cur.execute("ROLLBACK")
frame_count = self.frame_id
file_path = "/home/administrator/historical_videos/" + str(unique_id) + ".avi"
file_path = "/home/administrator/aarti_violation_videos/" + str(unique_id) + ".webm"
cur.execute(sql,
[frame_count, 4, str(unique_id), datetime.datetime.now(), file_path,
v])
......@@ -856,7 +856,7 @@ class Ppe(ModelWrapper):
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write(rec_inf[2] + ".webm")
f.write("\n")
del self.active_rec[object_id]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment