apps/frigate-rockchip
1
0
Fork 0
mirror of https://github.com/MarcA711/frigate-rockchip.git synced 2025-10-05 16:57:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

check plasma store and consolidate frame drawing

Browse Source
This commit is contained in:
Blake Blackshear
2020-02-16 08:00:41 -06:00
parent 569e07949f
commit 80b9652f7a
4 changed files with 54 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
detect_objects.py
View File

@@ -80,6 +80,11 @@ def main():
# start plasma store
plasma_cmd = ['plasma_store', '-m', '400000000', '-s', '/tmp/plasma']
plasma_process = sp.Popen(plasma_cmd, stdout=sp.DEVNULL, stderr=sp.DEVNULL)
time.sleep(1)
rc = plasma_process.poll()
if rc is not None:
raise RuntimeError("plasma_store exited unexpectedly with "
"code %d" % (rc,))
##
# Setup config defaults for cameras
@@ -95,6 +100,7 @@ def main():
# Start the shared tflite process
tflite_process = EdgeTPUProcess(MODEL_PATH)
# start the camera processes
camera_processes = []
camera_stats_values = {}
for name, config in CONFIG['cameras'].items():
@@ -167,9 +173,13 @@ def main():
while True:
# max out at 1 FPS
time.sleep(1)
frame = object_processor.current_frame_with_objects(camera_name)
frame = object_processor.get_current_frame(camera_name)
if frame is None:
frame = np.zeros((720,1280,3), np.uint8)
frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
ret, jpg = cv2.imencode('.jpg', frame)
yield (b'--frame\r\n'
b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n\r\n')
b'Content-Type: image/jpeg\r\n\r\n' + jpg.tobytes() + b'\r\n\r\n')
app.run(host='0.0.0.0', port=WEB_PORT, debug=False)