使用tensorflow object detection API实现目标检测

环境

Windows7 x64
conda 4.3.30

1、TensorFlow安装

首先在conda中创建TensorFlow环境

conda create -n tensorflow python=3.6.2

激活tensorflow环境

activate tensorflow

安装tensorflow

pip install tensorflow==1.12.0

安装完成后会在anaconda安装路径envs文件下自动创建tensorflow目录
在这里插入图片描述

2、tensorflow object detection API安装

下载安装包:https://github.com/xizhonghuai/tensorflow_object_detection_demo
在这里插入图片描述
下载完成后解压,将tensorflow_models文件夹拷贝到不含中文任意路径下,如:
在这里插入图片描述
在anaconda安装路径下找到刚才新创建的tensorflow环境目录
在这里插入图片描述
在\Lib\site-packages下创建tensorflow_model.pth文件,内容如下:

E:\python\workspace\tensorflow_models\research
E:\python\workspace\tensorflow_models\research\slim

进入E:\python\workspace\tensorflow_models\research\slim文件夹下分别执行一下命令:

python setup.py build 
python setup.py install

运行后如果出现error: could not create ‘build’,请删掉\slim文件夹下BUILD文件,在重新执行。

进入E:\python\workspace\tensorflow_models\research文件夹下分别执行一下命令:

python setup.py build 
python setup.py install

这个过程时间稍长,若出现报错情况,请仔细查看报错信息是否提示缺少相关依赖,手动通过pip安装。

测试tensorflow object detection API是否安装成功,在E:\python\workspace\tensorflow_models\research下执行

object_detection/builders/model_builder_test.py

无报错表示安装成功。

说明:
报错信息提示找不到相关模块,请卸载tensorflow,重新安装其他版本。
报错信息提示DLL相关错误,请卸载tensorflow,在以下链接中安装 https://github.com/fo40225/tensorflow-windows-wheel

3、测试

下载官方已训练好的模型到本地,这里我们使用ssd_mobilenet_v1_coco(模型包含90种常见物体的识别)
https://github.com/xizhonghuai/tensorflow_object_detection_demo/blob/master/tensorflow_models/research/object_detection/g3doc/detection_model_zoo.md

在这里插入图片描述
点击下载ssd_mobilenet_v1_coco模型你解压到本地。

打开iead,创建python工程,创建com包,在com包下分别创建,data、test_images、model文件夹

将刚才解压模型包中的frozen_inference_graph.pb文件拷贝到model文件下。

在E:\python\workspace\tensorflow_models\research\object_detection\data_back_up文件夹下拷贝mscoco_label_map.pbtxt文件到data文件下。

在网上随便找一张图片重命名为image1.jpg存放到test_images下

项目结构如下:
在这里插入图片描述

在com包下创建py代码:

import numpy as np
import os
import six.moves.urllib as urllib
import sys
import tarfile
import tensorflow as tf
import zipfile

from collections import defaultdict
from io import StringIO
# from matplotlib import pyplot as plt
from PIL import Image
## Env setup
# This is needed to display the images.

# This is needed since the notebook is stored in the object_detection folder.
sys.path.append("..")

from object_detection.utils  import label_map_util
from object_detection.utils  import visualization_utils as vis_util

os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"

# Model preparation
## Variables

#Any model exported using the `export_inference_graph.py` tool can be loaded here #simply by changing `PATH_TO_CKPT` to point to a new .pb file.
#By default we use an "SSD with Mobilenet" model here. See the [detection model zoo](https://github.com/tensorflow/models/blob/master/object_detection/g3doc/detection_mo#del_zoo.md) for a list of other models that can be run out-of-the-box with varying #speeds and accuracies.

# Path to frozen detection graph. This is the actual model that is used for the object detection.
PATH_TO_CKPT = 'model/frozen_inference_graph.pb'

PATH_TO_LABELS = os.path.join('data', 'mscoco_label_map.pbtxt')

NUM_CLASSES = 1


## Load a (frozen) Tensorflow model into memory.
detection_graph = tf.Graph()
with detection_graph.as_default():
    od_graph_def = tf.GraphDef()
    with tf.gfile.GFile(PATH_TO_CKPT, 'rb') as fid:
        print(PATH_TO_CKPT)
        serialized_graph = fid.read()
        od_graph_def.ParseFromString(serialized_graph)
        tf.import_graph_def(od_graph_def, name='')



label_map = label_map_util.load_labelmap(PATH_TO_LABELS)
categories = label_map_util.convert_label_map_to_categories(label_map, max_num_classes=NUM_CLASSES, use_display_name=True)
category_index = label_map_util.create_category_index(categories)


## Helper code
def load_image_into_numpy_array(image):
    (im_width, im_height) = image.size
    return np.array(image.getdata()).reshape(
        (im_height, im_width, 3)).astype(np.uint8)


# Detection
# For the sake of simplicity we will use only 2 images:
# image1.jpg
# image2.jpg
# If you want to test the code with your images, just add path to the images to the TEST_IMAGE_PATHS.
PATH_TO_TEST_IMAGES_DIR = 'test_images'
TEST_IMAGE_PATHS = [ os.path.join(PATH_TO_TEST_IMAGES_DIR, 'image{}.jpg'.format(i)) for i in range(1, 2) ]
#TEST_IMAGE_PATHS = ['test_images']
# Size, in inches, of the output images.
IMAGE_SIZE = (12, 8)


with detection_graph.as_default():
    with tf.Session(graph=detection_graph) as sess:
        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_into_numpy_array(image)
            # Expand dimensions since the model expects images to have shape: [1, None, None, 3]
            image_np_expanded = np.expand_dims(image_np, axis=0)
            image_tensor = detection_graph.get_tensor_by_name('image_tensor:0')
            # Each box represents a part of the image where a particular object was detected.
            boxes = detection_graph.get_tensor_by_name('detection_boxes:0')
            # Each score represent how level of confidence for each of the objects.
            # Score is shown on the result image, together with the class label.
            scores = detection_graph.get_tensor_by_name('detection_scores:0')
            classes = detection_graph.get_tensor_by_name('detection_classes:0')
            num_detections = detection_graph.get_tensor_by_name('num_detections:0')
            # Actual detection.
            (boxes, scores, classes, num_detections) = sess.run(
                [boxes, scores, classes, num_detections],
                feed_dict={image_tensor: image_np_expanded})
            # Visualization of the results of a detection.
            vis_util.visualize_boxes_and_labels_on_image_array(
                image_np,
                np.squeeze(boxes),
                np.squeeze(classes).astype(np.int32),
                np.squeeze(scores),
                category_index,
                use_normalized_coordinates=True,
                line_thickness=8)
            # plt.figure(figsize=IMAGE_SIZE)            # plt.imshow(image_np)
            # plt.show()

            im = Image.fromarray(image_np)
            im.save("ret.jpeg")
print("OK")

测试:

执行python代码,运行成功后将生成识别结果并保存图片到com包下。
在这里插入图片描述

在这里插入图片描述

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
TensorFlow Object Detection API 是一个开源项目,它提供了一系列基于 TensorFlow 的工具和库,用于实现目标检测任务。对于 macOS 系统,我们可以通过以下步骤来使用 TensorFlow Object Detection API: 1. 安装 TensorFlow:在 macOS 上安装 TensorFlow使用 TensorFlow Object Detection API 的前提。你可以通过 pip 命令进行安装,例如在终端中执行 `pip install tensorflow`。 2. 下载 TensorFlow Object Detection API:打开终端并导航到适合你的工作目录中,然后使用 git 命令来克隆 TensorFlow Object Detection API 的 GitHub 仓库,例如执行 `git clone https://github.com/tensorflow/models.git`。 3. 安装依赖项:进入克隆的模型目录中,找到 research 文件夹并进入。然后运行 `pip install -r object_detection/requirements.txt` 命令来安装所需的依赖项。 4. 下载预训练模型:在 TensorFlow Object Detection API 中,我们可以使用预训练的模型来进行目标检测。你可以从 TensorFlow Model Zoo 中下载适合你任务的模型,并将其解压到你的工作目录中。 5. 运行实例代码:在 research/object_detection 目录中,你可以找到一些示例代码,用于训练、评估和使用目标检测模型。可以通过阅读这些示例代码并根据自己的需求进行修改。例如,你可以使用 `python object_detection/builders/model_builder_tf2_test.py` 命令来运行一个模型的测试。 以上是在 macOS 上使用 TensorFlow Object Detection API 的基本步骤,你可以根据你的具体需求进行更多的深入研究和调整。希望这些信息能帮助到你!

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值