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):
size = (frame_width, frame_height)
working_dir = "output"
file_name =str(unique_id) + ".mp4"
file_name =str(unique_id) + ".avi"
file_name = os.path.join(working_dir, file_name)
# Below VideoWriter object will create
# a frame of above defined The output
# is stored in 'filename.avi' file.
fourcc = cv2.VideoWriter_fourcc(*'h264')
result = cv2.VideoWriter(file_name,
fourcc,
10, size)
cv2.VideoWriter_fourcc(*'MJPG'),
10, size)
# fourcc = cv2.VideoWriter_fourcc(*'h264')
# result = cv2.VideoWriter(file_name,
# fourcc,
# 10, size)
# fourcc = cv2.VideoWriter_fourcc(*'vp80')
# result = cv2.VideoWriter(file_name, fourcc, 20, (640, 480))
return result
......@@ -446,7 +449,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("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("\n")
del self.active_rec[object_id]
......@@ -530,7 +533,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("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("\n")
del self.active_rec[object_id]
......@@ -555,7 +558,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("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("\n")
del self.active_rec[object_id]
......@@ -639,7 +642,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("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("\n")
del self.active_rec[object_id]
......@@ -662,7 +665,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("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("\n")
del self.active_rec[object_id]
......@@ -748,7 +751,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("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("\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