统计图片test.py（基于tensorflow）

“““对图片进行检测 分类统计目标个数“““
import numpy as np
import os
import sys
import cv2
import time
import tensorflow as tf

from object_detection.utils import label_map_util
from object_detection.utils import visualization_utils as vis_util
class_keep = [3 5]
################################part2#######################################
#将训练好的模型放入内存
PATH_TO_CKPT =  ‘./MODEL_NAME/frozen_inference_graph.pb‘
#PATH_TO_CKPT =  ‘.//faster_rcnn_inception_resnet_v2_atrous_oid_2018_01_28//frozen_inference_graph.pb‘
PATH_TO_LABELS = ‘./MODEL_NAME/oid_bbox_trainable_label_map.pbtxt‘
#PATH_TO_LABELS = ‘.//faster_rcnn_inception_resnet_v2_atrous_oid_2018_01_28//oid_bbox_trainable_label_map.pbtxt‘

NUM_CLASSES = 90
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=‘‘）
#