-
大小: 5KB文件類型: .py金幣: 1下載: 0 次發布日期: 2021-01-03
- 語言: Python
- 標簽: opencv??tensorflow??物體檢測??
資源簡介
利用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 評論
共有 條評論