資源簡介
物體檢測識別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??ob
?????文件????????5042??2017-10-04?13:10??ob
?????目錄???????????0??2018-03-19?15:31??__MACOSX\
?????目錄???????????0??2018-03-19?15:31??__MACOSX\ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????文件????????1064??2017-10-04?13:10??ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????文件?????????958??2017-10-04?13:10??ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????文件????????5680??2017-10-04?13:10??ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????文件???????????0??2017-10-04?13:10??ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????目錄???????????0??2018-02-01?16:09??ob
?????文件?????????662??2017-10-04?13:10??ob
?????目錄???????????0??2018-03-19?15:31??__MACOSX\ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????文件???????????0??2017-10-04?13:10??ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????目錄???????????0??2018-02-01?16:09??ob
?????文件????????6690??2018-02-01?16:09??ob
?????文件?????????182??2018-02-01?16:09??ob
?????文件????????7886??2017-10-04?13:10??ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????文件?????????212??2018-02-01?16:09??__MACOSX\ob
?????文件????????1254??2017-10-04?13:10??ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????文件????????1061??2017-10-04?13:10??ob
?????文件?????????212??2017-10-04?13:10??__MACOSX\ob
?????目錄???????????0??2018-02-01?16:09??ob
?????目錄???????????0??2017-10-04?13:10??ob
............此處省略567個文件信息
評論
共有 條評論