Tensorflow object detection API输出识别对象的坐标等信息
此程序基于Tensorflow object detection API。
如果你不想安装API,但想使用此程序,请先完成
https://blog.csdn.net/weixin_43599336/article/details/83713790
视频演示地址:https://www.bilibili.com/video/av32418677/?p=3
# By Bend_Function
# https://space.bilibili.com/275177832
# 视频演示地址:[https://www.bilibili.com/video/av32418677/?p=3]
# 模型下载地址 https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md
# 可以放在任何文件夹下运行(前提正确配置API[环境变量])
import time
start = time.time()
import numpy as np
import os
import tensorflow as tf
from PIL import Image
from object_detection.utils import label_map_util
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
# 要改的内容
###############################################
PATH_TO_CKPT = 'model\\ssd_mobilenet_v1_graph.pb' # 模型及标签地址
PATH_TO_LABELS = 'model\\mscoco_label_map.pbtxt'
NUM_CLASSES = 90 # 检测对象个数
PATH_TO_TEST_IMAGES_DIR = os.getcwd()+'\\test_images' # 测试图片路径
confident = 0.5 # 置信度,即scores>confident的目标才被输出
###############################################
# Load a (frozen) Tensorflow model into memory.
detection_graph