Skip to content

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

added video rec feature

parent a54f55ea
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
scripts/ppe.py
View file @ f0337933
...@@ -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("recorded_video_list.txt", "a") as f: with open("/home/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("recorded_video_list.txt", "a") as f: with open("/home/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("recorded_video_list.txt", "a") as f: with open("/home/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("recorded_video_list.txt", "a") as f: with open("/home/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("recorded_video_list.txt", "a") as f: with open("/home/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("recorded_video_list.txt", "a") as f: with open("/home/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