ssd_mobilenet_demo

improt tensorflow as tf
hello = tf.constant(‘hello‘)
sess = tf.Session()
print(sess.run(hello))

当输出hello则装好tensorflow

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

当输出：
Sample Output
[name: "/cpu:0" device_type: "CPU" memory_limit: 
name: "/gpu:0" device_type: "GPU" .............GeForce GTX 1070

# From tensorflow/models/
protoc object_detection/protos/*.proto --python_out=.

在这一步有时候会出错，可以尝试把/*.proto 这部分改成文件夹下具体的文件名，一个一个试，每运行一个，文件夹下应该出现对应的.py结尾的文件。不报错即可。

python object_detection/builders/model_builder_test.py

# coding: utf-8
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

if tf.__version__ < ‘1.4.0‘:
  raise ImportError(‘Please upgrade your tensorflow installation to v1.4.* or later!‘)

# This is needed to display the images.
get_ipython().magic(‘matplotlib inline‘)

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

from utils import label_map_util
from utils import visualization_utils as vis_util

# 下载模型名，设置对应的参数
MODEL_NAME = ‘ssd_mobilenet_v1_coco_2017_11_17‘
MODEL_FILE = MODEL_NAME + ‘.tar.gz‘
DOWNLOAD_BASE = ‘http://download.tensorflow.org/models/object_detection/‘

# 训练好的模型，用来检测
PATH_TO_CKPT = MODEL_NAME + ‘/frozen_inference_graph.pb‘

# 标签文件，记录了哪些标签需要识别
PATH_TO_LABELS = os.path.join(‘data‘, ‘mscoco_label_map.pbtxt‘)

# 类别数目，根据实际修改
NUM_CLASSES = 90


# ## 下载上面说的模型（不用改）
opener = urllib.request.URLopener()
opener.retrieve(DOWNLOAD_BASE + MODEL_FILE, MODEL_FILE)
tar_file = tarfile.open(MODEL_FILE)
for file in tar_file.getmembers():
  file_name = os.path.basename(file.name)
  if ‘frozen_inference_graph.pb‘ in file_name:
    tar_file.extract(file, os.getcwd())


#将训练完的载入内存（不用改）
detection_graph = tf.Graph()
with detection_graph.as_default():
  od_graph_def = tf.GraphDef()
  with tf.gfile.GFile(PATH_TO_CKPT, ‘rb‘) as fid:
    serialized_graph = fid.read()
    od_graph_def.ParseFromString(serialized_graph)
    tf.import_graph_def(od_graph_def, name=‘‘)


# ## 载入标签map（不用改）
# Label maps map indices to category names, so that when our convolution network predicts `5`, 
we know that this corresponds to `airplane`.  
Here we use internal utility functions, but anything that returns a dictionary mapping integers to appropriate string labels would be fine

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)

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)


"""
检测部分
"""
# 目标文件夹名
PATH_TO_TEST_IMAGES_DIR = ‘test_images‘
# 源码中test_images文件夹下就两张image，名字分别为image1.jpg和image2.jpg
# 如果想用自己的image，有5张图片，分别为hello1.jpg.....hello5.jpg可以改成：
# TEST_IMAGE_PATHS = [ os.path.join(PATH_TO_TEST_IMAGES_DIR, ‘hello{}.jpg‘.format(i)) for i in range(1, 6) ]
TEST_IMAGE_PATHS = [ os.path.join(PATH_TO_TEST_IMAGES_DIR, ‘image{}.jpg‘.format(i)) for i in range(1, 3) ]

# 设置输出图像的英尺
IMAGE_SIZE = (12, 8)



#运行，进行检测
with detection_graph.as_default():
  with tf.Session(graph=detection_graph) as sess:
    # Definite input and output Tensors for detection_graph
    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.
    detection_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.
    detection_scores = detection_graph.get_tensor_by_name(‘detection_scores:0‘)
    detection_classes = detection_graph.get_tensor_by_name(‘detection_classes:0‘)
    num_detections = detection_graph.get_tensor_by_name(‘num_detections: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_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)
      # Actual detection.
      (boxes, scores, classes, num) = sess.run(
          [detection_boxes, detection_scores, detection_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)