“““對圖片進行檢測?分類統計目標個數“““
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=‘‘）
#