tensorflow目标检测代码

tensorflow目标检测代码，从摄像头或者视频源进行任务目标检测 python2.7 tensorflow

import numpy as np
import os
import six.moves.urllib as urllib
import sys
import tarfile
import tensorflow as tf
import zipfile
import multiprocessing
from multiprocessing import Queue
import time
import argparse

import cv2


from myutil import downloadutil fps_measure
from object_detection.utils import label_map_util
from object_detection.utils import visualization_utils as vis_util


arg_parser = argparse.ArgumentParser（）
arg_parser.add_argument（‘-v‘ ‘--video‘ type=str required=True
                        help=“video file for detection“）

args = arg_parser.parse_args（）

# What model to download.
MODEL_NAME = ‘ssd_mobilenet_v1_coco_11_06_2017‘


def load_graph（model_name=‘ssd_mobilenet_v1_coco_11_06_2017‘）:
    MODEL_FILE = MODEL_NAME + ‘.tar.gz‘
    DOWNLOAD_base = ‘http://download.tensorflow.org/models/object_detection/‘
    # Path to frozen detection graph. This is the actual model that is used for the object detection.
    PATH_TO_CKPT = MODEL_NAME + ‘/frozen_inference_graph.pb‘
    downloadutil.maybe_download（os.getcwd（） MODEL_FILE
                                DOWNLOAD_base+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（））
    # load graph
    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=‘‘）
    return detection_graph

NUM_CLASSES = 90

def load_label_map（label_map_name num_class）:
    # List of the strings that is used to add correct label for each box.
    PATH_TO_LABELS = os.path.join（‘data‘ label_map_name）
    # load label map
    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_class use_display_name=True）
    category_index = label_map_util.create_category_index（categories）
    return category_index

# def load_image_into_numpy_array（image）:
#     # import pdb; pdb.set_trace（）
#     # （im_width im_height） = image.size
#     （im_width im_height） = image.shape
#     # return np.array（image.getdata（））.reshape（
#     #                     （im_height im_width 3））.astype（np.uint8）
#     return image.reshape（（im_height im_width 3））.astype（np.uint8）

# def detect_object（detection_graph image category_index）:
#     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

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-10-26 07:36  object_detection_with_tensorflow-master\
     文件        1192  2017-10-26 07:36  object_detection_with_tensorflow-master\.gitignore
     目录           0  2017-10-26 07:36  object_detection_with_tensorflow-master\data\
     文件        5056  2017-10-26 07:36  object_detection_with_tensorflow-master\data\mscoco_label_map.pbtxt
     文件         705  2017-10-26 07:36  object_detection_with_tensorflow-master\data\pascal_label_map.pbtxt
     文件        1532  2017-10-26 07:36  object_detection_with_tensorflow-master\data\pet_label_map.pbtxt
     目录           0  2017-10-26 07:36  object_detection_with_tensorflow-master\images\
     文件     1827353  2017-10-26 07:36  object_detection_with_tensorflow-master\images\bigbang1.png
     文件     1512000  2017-10-26 07:36  object_detection_with_tensorflow-master\images\bigbang2.png
     文件      876595  2017-10-26 07:36  object_detection_with_tensorflow-master\images\bigbang3_fps.png
     文件        1071  2017-10-26 07:36  object_detection_with_tensorflow-master\LICENSE
     目录           0  2017-10-26 07:36  object_detection_with_tensorflow-master\myutil\
     文件         733  2017-10-26 07:36  object_detection_with_tensorflow-master\myutil\downloadutil.py
     文件         543  2017-10-26 07:36  object_detection_with_tensorflow-master\myutil\downloadutil_test.py
     文件         427  2017-10-26 07:36  object_detection_with_tensorflow-master\myutil\fps_measure.py
     目录           0  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\
     目录           0  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\anchor_generators\
     文件        1267  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\anchor_generators\BUILD
     文件        8499  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\anchor_generators\grid_anchor_generator.py
     文件        2950  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\anchor_generators\grid_anchor_generator_test.py
     文件       12824  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\anchor_generators\multiple_grid_anchor_generator.py
     文件       11513  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\anchor_generators\multiple_grid_anchor_generator_test.py
     文件           0  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\anchor_generators\__init__.py
     目录           0  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\box_coders\
     文件        2281  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\box_coders\BUILD
     文件        3837  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\box_coders\faster_rcnn_box_coder.py
     文件        4213  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\box_coders\faster_rcnn_box_coder_test.py
     文件        6346  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\box_coders\keypoint_box_coder.py
     文件        5900  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\box_coders\keypoint_box_coder_test.py
     文件        2364  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\box_coders\mean_stddev_box_coder.py
     文件        2396  2017-10-26 07:36  object_detection_with_tensorflow-master\object_detection\box_coders\mean_stddev_box_coder_test.py
............此处省略261个文件信息