Tensorflow object detection API输出识别对象的坐标等信息

最新推荐文章于 2023-05-26 15:07:22 发布

Bend_Function

最新推荐文章于 2023-05-26 15:07:22 发布

阅读量3.1k

点赞数 4

分类专栏： Object Detection API使用文章标签： Tensorflow 目标检测

本文链接：https://blog.csdn.net/weixin_43599336/article/details/84112134

版权

本文介绍如何利用Tensorflow Object Detection API来获取识别对象的坐标信息。程序基于该API，若要运行，需参考指定链接完成设置。模型下载链接和视频演示也可供参考。

摘要由CSDN通过智能技术生成

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