apps/frigate-rockchip
1
0
Fork 0
mirror of https://github.com/MarcA711/frigate-rockchip.git synced 2025-11-02 21:14:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

only save images of persons

Browse Source
This commit is contained in:
blakeblackshear
2019-02-22 21:03:47 -06:00
parent 9486a321c6
commit 5258803cd7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
detect_objects.py
View File

@@ -63,7 +63,7 @@ def detect_objects(cropped_frame, sess, detection_graph, region_size, region_x_o
feed_dict={image_tensor: image_np_expanded})
if debug:
if len([category_index.get(value) for index,value in enumerate(classes[0]) if scores[0,index] > 0.5]) > 0:
if len([value for index,value in enumerate(classes[0]) if str(category_index.get(value)) == 'person' and scores[0,index] > 0.5]) > 0:
vis_util.visualize_boxes_and_labels_on_image_array(
cropped_frame,
np.squeeze(boxes),