Commit 6e046113 authored by Sikhin VC's avatar Sikhin VC

added video rec feature

parent 1d8b80e7
...@@ -351,7 +351,7 @@ class Ppe(ModelWrapper): ...@@ -351,7 +351,7 @@ class Ppe(ModelWrapper):
size = (frame_width, frame_height) size = (frame_width, frame_height)
working_dir = os.getcwd() working_dir = os.getcwd()
file_name =str(unique_id) + ".avi" 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 # Below VideoWriter object will create
# a frame of above defined The output # a frame of above defined The output
# is stored in 'filename.avi' file. # is stored in 'filename.avi' file.
...@@ -443,7 +443,7 @@ class Ppe(ModelWrapper): ...@@ -443,7 +443,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list: for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list, self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2]) 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(rec_inf[2] + ".avi")
f.write("\n") f.write("\n")
del self.active_rec[object_id] del self.active_rec[object_id]
...@@ -527,7 +527,7 @@ class Ppe(ModelWrapper): ...@@ -527,7 +527,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list: for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list, self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2]) 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(rec_inf[2] + ".avi")
f.write("\n") f.write("\n")
del self.active_rec[object_id] del self.active_rec[object_id]
...@@ -552,7 +552,7 @@ class Ppe(ModelWrapper): ...@@ -552,7 +552,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list: for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list, self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2]) 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(rec_inf[2] + ".avi")
f.write("\n") f.write("\n")
del self.active_rec[object_id] del self.active_rec[object_id]
...@@ -636,7 +636,7 @@ class Ppe(ModelWrapper): ...@@ -636,7 +636,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list: for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list, self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2]) 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(rec_inf[2] + ".avi")
f.write("\n") f.write("\n")
del self.active_rec[object_id] del self.active_rec[object_id]
...@@ -659,7 +659,7 @@ class Ppe(ModelWrapper): ...@@ -659,7 +659,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list: for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list, self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2]) 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(rec_inf[2] + ".avi")
f.write("\n") f.write("\n")
del self.active_rec[object_id] del self.active_rec[object_id]
...@@ -745,7 +745,7 @@ class Ppe(ModelWrapper): ...@@ -745,7 +745,7 @@ class Ppe(ModelWrapper):
for v in temp_v_list: for v in temp_v_list:
self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list, self.send_payload(frame=resize_to_64_64(frame), message=temp_v_list,
event=v, frame_id=rec_inf[2]) 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(rec_inf[2] + ".avi")
f.write("\n") f.write("\n")
del self.active_rec[object_id] 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