0 前言
在前面我们将utils文件夹中的visualization_utils进行了改造,现在将检测出来的数量输出乘列表的格式,以便后续的数据处理
1 改造visualization_utils
打开visualization_utils这个文件,大约在700行左右的地方,将return image改成return counterBoxs
2 改变自带的物体检测代码
打开自带的物体检测模型,将最后一组代码改变,红框位置为改变的地方,此处改变的改变的意思就是将每一个输出的结果都放入data的列表中,并且每3个元素就计算依次最大值。
代码附上~
data = []
i = 0
range_number = 0
for image_path in TEST_IMAGE_PATHS:
image = Image.open(image_path)
# the array based representation of the image will be used later in order to prepare the
# result image with boxes and labels on it.
image_np = load_image_in