Skip to content

A
aart_ppe_detection
Commit 6e046113 authored by Sikhin VC's avatar Sikhin VC
Browse files
Options

added video rec feature

parent 1d8b80e7
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
scripts/ppe.py
View file @ 6e046113
......@@ -351,7 +351,7 @@ class Ppe(ModelWrapper):
size = (frame_width, frame_height)
working_dir = os.getcwd()
file_name =str(unique_id) + ".avi"
file_name = os.path.join("/home/output", file_name)
file_name = os.path.join(working_dir, "output", file_name)
# Below VideoWriter object will create
# a frame of above defined The output
# is stored in 'filename.avi' file.
......@@ -443,7 +443,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("/home/output/recorded_video_list.txt", "a") as f:
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write("\n")
del self.active_rec[object_id]
......@@ -527,7 +527,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("/home/output/recorded_video_list.txt", "a") as f:
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write("\n")
del self.active_rec[object_id]
......@@ -552,7 +552,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("/home/output/recorded_video_list.txt", "a") as f:
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write("\n")
del self.active_rec[object_id]
......@@ -636,7 +636,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("/home/output/recorded_video_list.txt", "a") as f:
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write("\n")
del self.active_rec[object_id]
......@@ -659,7 +659,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("/home/output/recorded_video_list.txt", "a") as f:
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
f.write("\n")
del self.active_rec[object_id]
......@@ -745,7 +745,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2])
with open("/home/output/recorded_video_list.txt", "a") as f:
with open("output/recorded_video_list.txt", "a") as f:
f.write(rec_inf[2] + ".avi")
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