Tensorflow对象检测choose/switchclass检测

Tensorflow object detection choose/switch class to detect

显示如何选择一个class进行检测。您如何更改代码以打开显示的 classes?示例过滤器只显示人,然后更改为只显示汽车等。

假设你有分数并且 classes:

import numpy as np

scores = sorted(np.random.rand(100), reverse=True)
classes = np.random.randint(0, 4, 100)

你可以把不是class的概率变成0:

modified_scores = [a if b == 1 else 0 for a, b in zip(scores, classes)]

它不会显示 1) 低于阈值和 2) 不符合所需 class 的 classes。