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