问答

TensorFlow的tensor如何转换为ndarray?

作者:admin 2021-09-17 我要评论

yolo_model.predict返回的是tensor,但我后续做cpu_nms时要求输入的是ndarray类型,我尝试过用.eval()(可能是我写的不对?),但是会出现FailedPreconditionErr...

在说正事之前,我要推荐一个福利:你还在原价购买阿里云、腾讯云、华为云服务器吗?那太亏啦!来这里,新购、升级、续费都打折,能够为您省60%的钱呢!2核4G企业级云服务器低至69元/年,点击进去看看吧>>>)

yolo_model.predict返回的是tensor,但我后续做cpu_nms时要求输入的是ndarray类型,我尝试过用.eval()(可能是我写的不对?),但是会出现FailedPreconditionError的错误

img = np.asarray(img, np.float32)
img = img[np.newaxis, :] / 255.


with tf.Session() as sess:

    input_data = tf.placeholder(tf.float32, [1, args.new_size[1], args.new_size[0], 3], name='input_data')
    yolo_model = yolov3(args.num_class, args.anchors)
    with tf.variable_scope('yolov3'):
        pred_feature_maps = yolo_model.forward(input_data, False)

    pred_boxes, pred_confs, pred_probs = yolo_model.predict(pred_feature_maps)

    pred_scores = pred_confs * pred_probs
    # pred_boxes = pred_boxes.eval()
    # pred_scores = pred_scores.eval()

    boxes, scores, labels = cpu_nms(pred_boxes, pred_scores, args.num_class, max_boxes=200, score_thresh=0.3, iou_thresh=0.45)

    saver = tf.train.Saver()
    saver.restore(sess, args.restore_path)

    boxes_, scores_, labels_ = sess.run([boxes, scores, labels], feed_dict={input_data: img})

谢谢

版权声明:本文转载自网络,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。本站转载出于传播更多优秀技术知识之目的,如有侵权请联系QQ/微信:153890879删除

相关文章
  • TensorFlow的tensor如何转换为ndarray

    TensorFlow的tensor如何转换为ndarray

  • 关于多线程中的wait/notify机制

    关于多线程中的wait/notify机制

  • vue+elementui $t('action.comfirm')这

    vue+elementui $t('action.comfirm')这

  • 请问Laravel-admin Collapse组件如何设

    请问Laravel-admin Collapse组件如何设

腾讯云代理商
海外云服务器