TensorRT с TensorFlow не дает результата при выводе - PullRequest
Новая
       48

TensorRT с TensorFlow не дает результата при выводе

0 голосов
/ 19 января 2019

Я хочу использовать тензор для ускорения вывода модели тензорного потока.Но я не получил никакого результата, когда сделаю это, вот мой код.Ящики для возврата и оценки все Нет.Я использую TensorRT и tenorflow неправильно? 

import tensorflow as tf
import tensorflow.contrib.tensorrt as trt
import os
from PIL import Image
import numpy as np

with tf.Session() as sess:
    with tf.gfile.GFile('model/ssd_inceptionv2.pb',  'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
    trt_graph = trt.create_inference_graph(
            input_graph_def=graph_def,
            outputs=["detection_boxes", "detection_classes"])
    output_node = tf.import_graph_def(
            trt_graph,
            return_elements=["detection_boxes", "detecction_classes"], name='')
    input = tf.get_default_graph().get_tensor_by_name('image_tensor:0')

    for root, dirs, files in os.walk('dataset/test'):
        for f in files:
            if not f.endswith('.jpg'):
                continue
            print f
            image = Image.open(os.path.join(root, f))
            print image
            image = image.resize((300, 300), Image.ANTIALIAS)
            im_width, im_height = image.size
            image = np.array(image.getdata()).reshape((im_height, im_width, 3)).astype(np.uint8)
            image = np.expand_dims(image, axis=0)
            boxes, scores = sess.run(output_node, feed_dict={input: image})
            boxes, scores = np.squeeze(boxes), np.squeeze(scores)
            print boxes, scores
            for i, s in enumerate(scores):
                if s > 0.5:
                    print boxes[i], s

1 Ответ

0 голосов
/ 18 июля 2019

Проблема в определении output_node. Элементы return_elements должны быть определены как return_elments=["detection_boxes:0", "detection_classes:0"]

...