Print out mask file read failure

Added print statement when mask file could not be read.
This commit is contained in:
Christopher Svensson 2020-07-28 10:31:30 +02:00 committed by GitHub
parent 53ccc903da
commit 9e0601c090
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -190,6 +190,8 @@ def track_camera(name, config, global_objects_config, frame_queue, frame_shape,
# load in the mask for object detection
if 'mask' in config:
mask = cv2.imread("/config/{}".format(config['mask']), cv2.IMREAD_GRAYSCALE)
if img.size == 0:
print(f"Failed to read mask file")
else:
mask = None
@ -370,4 +372,4 @@ def track_camera(name, config, global_objects_config, frame_queue, frame_shape,
# add to the queue
detected_objects_queue.put((name, frame_time, object_tracker.tracked_objects))
print(f"{name}: exiting subprocess")
print(f"{name}: exiting subprocess")