@Author：Runsen

tfhub是tensorflow手官打造体能训练方法好的模式化的其中一个库房。昨天，我运行 tfhub中的模式化EfficientDet-Lite2 去相亲对象判断

首选的仿真建模 是EfficientDet-Lite2 喜欢的人的检测仿真建模 。它在具有着 91 个差异tag标签的 COCO17 数据表格集奋发向上行了的训练，并造成 TFLite 选用安装文件实施了优化调整。 EfficientDet-Lite 是一种系列的对运动/物上网网融洽的另一半监测3d模型。 迄今为止仍然未能随时日常访问浏览 //tfhub.dev，能能进行镜像软件 //hub.tensorflow.google.cn

加载TensorFlow 模型

圖片輸入：规格可变气门正时的三通管件道图片。輸入张量是一个个 tf.uint8兼有样式张量[None, height, width, 3]。

加载图像并将其处理为 TensorFlow 模型的tensor格式。

import tensorflow_hub as hub
import cv2
import numpy
import pandas as pd
import tensorflow as tf
import matplotlib.pyplot as plt

width = 1028
height = 1028

img = cv2.imread('image.jpg')
inp = cv2.resize(img, (width , height ))
rgb = cv2.cvtColor(inp, cv2.COLOR_BGR2RGB)
rgb_tensor = tf.convert_to_tensor(rgb, dtype=tf.uint8)
rgb_tensor = tf.expand_dims(rgb_tensor , 0)

打开EfficientDet-Lite2，逛网给予打开代码怎么用

考虑到下载百度模式化上行带宽没法，将设计区域转到colab中。

detector = hub.load("//hub.tensorflow.google.cn/tensorflow/efficientdet/lite2/detection/1")
boxes, scores, classes, num_detections = detector(boxes, scores, classes, num_detections = detector(rgb_tensor))

读取字典构成： detection_boxes：一个tf.float32形状的张量[N, 4]含有以下列顺序边界框坐标：[ymin, xmin, ymax, xmax]。detection_scores：包含检测分数tf.float32的形状张量[N]。detection_classes：包含标签文件中检测类索引tf.int的形状张量[N]。num_detections：一个tf.int只有一个值的张量，检测次数[N]。 接下是验测的scores和class

物体编号：47，72，77检测的置信度比较高。

label = ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus',
       'train', 'truck', 'boat', 'traffic light', 'fire hydrant', '-',
       'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog',
       'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra', 'giraffe',
       '-', 'backpack', 'umbrella', '-', '-', 'handbag', 'tie',
       'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball', 'kite',
       'baseball bat', 'baseball glove', 'skateboard', 'surfboard',
       'tennis racket', 'bottle', '-', 'wine glass', 'cup', 'fork',
       'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange',
       'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair',
       'couch', 'potted plant', 'bed', '-', 'dining table', '-', '-',
       'toilet', '-', 'tv', 'laptop', 'mouse', 'remote', 'keyboard',
       'cell phone', 'microwave', 'oven', 'toaster', 'sink',
       'refrigerator', '-', 'book', 'clock', 'vase', 'scissors',
       'teddy bear', 'hair drier', 'toothbrush', '-']

以下将tensor改变为numpy()

pred_labels = classes.numpy().astype('int')[0] 
pred_labels = [labels[i] for i in pred_labels]
pred_boxes = boxes.numpy()[0].astype('int')
pred_scores = scores.numpy()[0]

在最后设制一两个阀值，将加测框加到图片文字上。

for score, (ymin,xmin,ymax,xmax), label in zip(pred_scores, pred_boxes, pred_labels):
        if score < 0.5:
            continue
        score_txt = f'{100 * round(score)}%'
        img_boxes = cv2.rectangle(rgb,(xmin, ymax),(xmax, ymin),(0,255,0),2)      
        font = cv2.FONT_HERSHEY_SIMPLEX
        cv2.putText(img_boxes, label,(xmin, ymax-10), font, 0.5, (255,0,0), 2, cv2.LINE_AA)
        cv2.putText(img_boxes,score_txt,(xmax, ymax-10), font, 0.5, (255,0,0), 2, cv2.LINE_AA)
plt.figure(figsize=(10,10))
plt.imshow(img_boxes)
plt.savefig('image_pred.jpg',transparent=True, )