Skip to content

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

added video recording format

parent 8e4fecfd
Hide whitespace changes
Inline Side-by-side
Showing with 13 additions and 10 deletions
scripts/ppe.py
View file @ 5b7729b4
...@@ -350,15 +350,18 @@ class Ppe(ModelWrapper): ...@@ -350,15 +350,18 @@ class Ppe(ModelWrapper):
size = (frame_width, frame_height) size = (frame_width, frame_height)
working_dir = "output" working_dir = "output"
file_name =str(unique_id) + ".mp4" file_name =str(unique_id) + ".avi"
file_name = os.path.join(working_dir, file_name) file_name = os.path.join(working_dir, 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.
fourcc = cv2.VideoWriter_fourcc(*'h264')
result = cv2.VideoWriter(file_name, result = cv2.VideoWriter(file_name,
fourcc, cv2.VideoWriter_fourcc(*'MJPG'),
10, size) 10, size)
# fourcc = cv2.VideoWriter_fourcc(*'h264')
# result = cv2.VideoWriter(file_name,
# fourcc,
# 10, size)
# fourcc = cv2.VideoWriter_fourcc(*'vp80') # fourcc = cv2.VideoWriter_fourcc(*'vp80')
# result = cv2.VideoWriter(file_name, fourcc, 20, (640, 480)) # result = cv2.VideoWriter(file_name, fourcc, 20, (640, 480))
return result return result
...@@ -446,7 +449,7 @@ class Ppe(ModelWrapper): ...@@ -446,7 +449,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("output/recorded_video_list.txt", "a") as f: with open("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]
...@@ -530,7 +533,7 @@ class Ppe(ModelWrapper): ...@@ -530,7 +533,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("output/recorded_video_list.txt", "a") as f: with open("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]
...@@ -555,7 +558,7 @@ class Ppe(ModelWrapper): ...@@ -555,7 +558,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("output/recorded_video_list.txt", "a") as f: with open("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]
...@@ -639,7 +642,7 @@ class Ppe(ModelWrapper): ...@@ -639,7 +642,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("output/recorded_video_list.txt", "a") as f: with open("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]
...@@ -662,7 +665,7 @@ class Ppe(ModelWrapper): ...@@ -662,7 +665,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("output/recorded_video_list.txt", "a") as f: with open("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]
...@@ -748,7 +751,7 @@ class Ppe(ModelWrapper): ...@@ -748,7 +751,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("output/recorded_video_list.txt", "a") as f: with open("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