0 前言
在前面我们将utils文件夹中的visualization_utils进行了改造,现在将检测出来的数量输出乘列表的格式,以便后续的数据处理
1改造visualization_utils
PS:前面处理视频的数据输出(见我的AI(十))改造过了visualization_utils就不需要再改造了
打开visualization_utils这个文件,大约在700行左右的地方,将return image改成return counterBoxs
2 改变实时的物体检测代码
将实时的物体检测代码进行改造,完整代码附上,可以去pycham上直接运行
import numpy as np
import tensorflow as tf
import cv2
import os
from object_detection.utils import visualization_utils as vis_util
from object_detection.utils import label_map_util
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
cv2.setUseOptimized(True) # 加速cv
# 要改的内容
###############################################
PATH_TO_CKPT = 'C:\\Machine_learning\\mo