类态张量流目标检测算法

Question

我们可以在Tensorflow对象检测API类中计算检测到的对象吗？假设你有两个类car和bicyle，我想按类计算每个对象ex，车数：5辆自行车计数：3。你知道吗

Answer 1

实现这一点的一种方法是执行以下操作

boxes = detection_graph.get_tensor_by_name('detection_boxes:0')
scores = detection_graph.get_tensor_by_name('detection_scores:0')
classes = detection_graph.get_tensor_by_name('detection_classes:0')
num_detections = detection_graph.get_tensor_by_name('num_detections:0')
(boxes, scores, classes, num_detections) = sess.run([boxes, scores, classes, num_detections],feed_dict={image_tensor: image_np_expanded})

final_score = np.squeeze(scores)    
car_count = 0

for i in range(100):
    if final_score[i] > 0.5:
        detected_class = int(classes[0][i])
        if detected_class == 1:
            car_count += 1

在这里，类名和编号根据您在label_map.pbtxt文件中指定的值进行映射。我希望这对你有帮助。如果你遇到任何问题，请告诉我