物体检测识别python源代码

物体检测识别python源代码，可以识别很多类物体，源码简洁

import os
import cv2
import time
import argparse
import multiprocessing
import numpy as np
import tensorflow as tf

from utils.app_utils import FPS WebcamVideoStream
from multiprocessing import Queue Pool
from object_detection.utils import label_map_util
from object_detection.utils import visualization_utils as vis_util

CWD_PATH = os.getcwd（）

# Path to frozen detection graph. This is the actual model that is used for the object detection.
MODEL_NAME = ‘ssd_mobilenet_v1_coco_11_06_2017‘
PATH_TO_CKPT = os.path.join（CWD_PATH ‘object_detection‘ MODEL_NAME ‘frozen_inference_graph.pb‘）

# List of the strings that is used to add correct label for each box.
PATH_TO_LABELS = os.path.join（CWD_PATH ‘object_detection‘ ‘data‘ ‘mscoco_label_map.pbtxt‘）

NUM_CLASSES = 90

# Loading 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_CLASSES
                                                            use_display_name=True）
category_index = label_map_util.create_category_index（categories）


def detect_objects（image_np sess detection_graph）:
    # 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）
    return image_np


def worker（input_q output_q）:
    # 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:
            serialized_graph = fid.read（）
            od_graph_def.ParseFromString（serialized_graph）
            tf.import_graph_def（od_graph_def name=‘‘）

        sess = tf.Session（graph=detection_graph）

    fps = FPS（）.start（）
    while True:
        fps.update（）
        frame = input_q.get（）
        frame_rgb = cv2.cvtCo

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2018-02-01 16:07  object_detector_app-master\
     文件        5042  2017-10-04 13:10  object_detector_app-master\object_detection_app.py
     目录           0  2018-03-19 15:31  __MACOSX\
     目录           0  2018-03-19 15:31  __MACOSX\object_detector_app-master\
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\._object_detection_app.py
     文件        1064  2017-10-04 13:10  object_detector_app-master\LICENSE
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\._LICENSE
     文件         958  2017-10-04 13:10  object_detector_app-master\environment.yml
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\._environment.yml
     文件        5680  2017-10-04 13:10  object_detector_app-master\object_detection_multithreading.py
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\._object_detection_multithreading.py
     文件           0  2017-10-04 13:10  object_detector_app-master\__init__.py
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\.___init__.py
     目录           0  2018-02-01 16:09  object_detector_app-master\utils\
     文件         662  2017-10-04 13:10  object_detector_app-master\utils\test_app_utils.py
     目录           0  2018-03-19 15:31  __MACOSX\object_detector_app-master\utils\
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\utils\._test_app_utils.py
     文件           0  2017-10-04 13:10  object_detector_app-master\utils\__init__.py
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\utils\.___init__.py
     目录           0  2018-02-01 16:09  object_detector_app-master\utils\__pycache__\
     文件        6690  2018-02-01 16:09  object_detector_app-master\utils\__pycache__\app_utils.cpython-36.pyc
     文件         182  2018-02-01 16:09  object_detector_app-master\utils\__pycache__\__init__.cpython-36.pyc
     文件        7886  2017-10-04 13:10  object_detector_app-master\utils\app_utils.py
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\utils\._app_utils.py
     文件         212  2018-02-01 16:09  __MACOSX\object_detector_app-master\._utils
     文件        1254  2017-10-04 13:10  object_detector_app-master\README.md
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\._README.md
     文件        1061  2017-10-04 13:10  object_detector_app-master\.gitignore
     文件         212  2017-10-04 13:10  __MACOSX\object_detector_app-master\._.gitignore
     目录           0  2018-02-01 16:09  object_detector_app-master\object_detection\
     目录           0  2017-10-04 13:10  object_detector_app-master\object_detection\test_images\
............此处省略567个文件信息