PROSAGA码农传奇-模式识别-如何使用tensorflow对象检测来仅检测人？

如何使用tensorflow对象检测来仅检测人？

作者: CC-f
发布时间: 2023-09-22 02:25:12 (9月前)
转自：

<div class =“post-text”itemprop =“text”>
  
    我看到你在线上使用了一个过滤器
     <code>
 b = [x for x in classes if x == 1]
 </code>
     只是得到所有的人检测。 （在标签地图中，人的身份正好是1）。但它没有用，因为你需要改变
     <code>
 boxes
 </code>
    ，
     <code>
 scores
 </code>
     和
     <code>
 classes
 </code>
     因此。试试这个 ：
  
  
    首先删除该行
  
   <pre>
 <code>
 b = [x for x in classes if x == 1]

</code>
 </pre>
  
    然后添加以下内容
     <code>
 sess.run()
 </code>
     功能
  
   <pre>
 <code>
 boxes = np.squeeze(boxes)
scores = np.squeeze(scores)
classes = np.squeeze(classes)

indices = np.argwhere(classes == 1)
boxes = np.squeeze(boxes[indices])
scores = np.squeeze(scores[indices])
classes = np.squeeze(classes[indices])

</code>
 </pre>
  
    然后调用可视化功能
  
   <pre>
 <code>
 vis_util.visualize_boxes_and_labels_on_image_array(
 image_np,
 boxes,
 classes,
 scores,
 category_index,
 use_normalized_coordinates=True,
 line_thickness=8)

</code>
 </pre>
  
    这个想法是模型可以产生多个类的检测，但只选择类人来对图像进行可视化。
  
</DIV>