Skip to content

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

added video recording format

parent 6e046113
Hide whitespace changes
Inline Side-by-side
Showing with 1180 additions and 13 deletions
app.py
View file @ 8c22f4bc
import os import os
# os.environ["config"]="{\"TZ\": \"Asia/Kolkata\", \"MONGO_URI\": \"mongodb://svc-ilens:svc2345@192.168.3.220:21017\", \"MONGO_DATABASE\": \"ilens_wps\", \"MONGO_COLLECTION\": \"janusDeployment\", \"MONGO_KEY\": \"deploymentId\", \"MONGO_VALUE\": \"rahul_12345\", \"MONGO_COLL\": \"serviceConfiguration\", \"MONGO_DB\": \"ilens_wps\"}" # os.environ["config"]="{\"TZ\": \"Asia/Kolkata\", \"MONGO_URI\": \"mongodb://svc-ilens:svc2345@192.168.3.220:21017\", \"MONGO_DATABASE\": \"ilens_wps\", \"MONGO_COLLECTION\": \"janusDeployment\", \"MONGO_KEY\": \"deploymentId\", \"MONGO_VALUE\": \"rahul_12345\", \"MONGO_COLL\": \"serviceConfiguration\", \"MONGO_DB\": \"ilens_wps\"}"
# os.environ["config"]="{\"TZ\": \"Asia/Kolkata\", \"MONGO_URI\": \"mongodb://admin:iLens$HPCLv605@10.5.2.91:2717\", \"MONGO_DATABASE\": \"ilens_ai\", \"MONGO_COLLECTION\": \"janusDeployment\", \"MONGO_KEY\": \"deploymentId\", \"MONGO_VALUE\": \"hpcl_cctv_ppe\", \"MONGO_COLL\": \"serviceConfiguration\", \"MONGO_DB\": \"ilens_ai\"}" # os.environ["config"]="{\"TZ\": \"Asia/Kolkata\", \"MONGO_URI\": \"mongodb://admin:iLens$HPCLv605@10.5.2.91:2717\", \"MONGO_DATABASE\": \"ilens_ai\", \"MONGO_COLLECTION\": \"janusDeployment\", \"MONGO_KEY\": \"deploymentId\", \"MONGO_VALUE\": \"hpcl_cctv_ppe\", \"MONGO_COLL\": \"serviceConfiguration\", \"MONGO_DB\": \"ilens_ai\"}"
# os.environ["config"]="{\"TZ\": \"Asia/Kolkata\", \"MONGO_URI\": \"mongodb://admin:iLens$1234@192.168.3.181:2717/admin\", \"MONGO_DATABASE\": \"ilens_ai\", \"MONGO_COLLECTION\": \"janusDeployment\", \"MONGO_KEY\": \"deploymentId\", \"MONGO_VALUE\": \"aarti_ppe\", \"MONGO_COLL\": \"serviceConfiguration\", \"MONGO_DB\": \"ilens_ai\"}" os.environ["config"]="{\"TZ\": \"Asia/Kolkata\", \"MONGO_URI\": \"mongodb://admin:iLens$1234@192.168.3.181:2717/admin\", \"MONGO_DATABASE\": \"ilens_ai\", \"MONGO_COLLECTION\": \"janusDeployment\", \"MONGO_KEY\": \"deploymentId\", \"MONGO_VALUE\": \"aarti_ppe\", \"MONGO_COLL\": \"serviceConfiguration\", \"MONGO_DB\": \"ilens_ai\"}"
from edge_engine.edge_processor import ExecutePipeline from edge_engine.edge_processor import ExecutePipeline
from edge_engine.edge_processor import Pubs from edge_engine.edge_processor import Pubs
from scripts import Ppe from scripts import Ppe
......
logs/ilens-edge_engine.log
View file @ 8c22f4bc
This source diff could not be displayed because it is too large. You can view the blob instead.
scripts/ppe.py
View file @ 8c22f4bc
...@@ -349,15 +349,18 @@ class Ppe(ModelWrapper): ...@@ -349,15 +349,18 @@ class Ppe(ModelWrapper):
frame_height = int(frame.shape[0]) frame_height = int(frame.shape[0])
size = (frame_width, frame_height) size = (frame_width, frame_height)
working_dir = os.getcwd() working_dir = "/home/administrator/aarti_violation_videos"
file_name =str(unique_id) + ".avi" file_name =str(unique_id) + ".mp4"
file_name = os.path.join(working_dir, "output", 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,
cv2.VideoWriter_fourcc(*'H264'), fourcc,
10, size) 10, size)
# fourcc = cv2.VideoWriter_fourcc(*'vp80')
# result = cv2.VideoWriter(file_name, fourcc, 20, (640, 480))
return result return result
def ppe_detection(self, frame, bbox, detection_objects, class_name, other_class_name, other_centroid): def ppe_detection(self, frame, bbox, detection_objects, class_name, other_class_name, other_centroid):
...@@ -443,7 +446,7 @@ class Ppe(ModelWrapper): ...@@ -443,7 +446,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("/home/administrator/aarti_violation_videos/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 +530,7 @@ class Ppe(ModelWrapper): ...@@ -527,7 +530,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("/home/administrator/aarti_violation_videos/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 +555,7 @@ class Ppe(ModelWrapper): ...@@ -552,7 +555,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("/home/administrator/aarti_violation_videos/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 +639,7 @@ class Ppe(ModelWrapper): ...@@ -636,7 +639,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("/home/administrator/aarti_violation_videos/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 +662,7 @@ class Ppe(ModelWrapper): ...@@ -659,7 +662,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("/home/administrator/aarti_violation_videos/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 +748,7 @@ class Ppe(ModelWrapper): ...@@ -745,7 +748,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("/home/administrator/aarti_violation_videos/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]
...@@ -962,8 +965,8 @@ class Ppe(ModelWrapper): ...@@ -962,8 +965,8 @@ class Ppe(ModelWrapper):
# frame = cv2.polylines(frame, self.polygon, # frame = cv2.polylines(frame, self.polygon,
# True, (255, 255, 255), 2) # True, (255, 255, 255), 2)
bbox, frame, class_name, other_class_name, other_centroid = self.inference(frame, class_list) bbox, frame, class_name, other_class_name, other_centroid = self.inference(frame, class_list)
# cv2.imshow("out", cv2.resize(frame, (900, 600))) cv2.imshow("out", cv2.resize(frame, (900, 600)))
# cv2.waitKey(1) cv2.waitKey(1)
# frame = self.draw_line_over_image(frame) # frame = self.draw_line_over_image(frame)
# if [True for e in dets if e['class'] == 'cement_bag']: # if [True for e in dets if e['class'] == 'cement_bag']:
......
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