发新帖

图像添加标注框总是失败

[复制链接]
323 2

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
  1. # -*- coding: utf-8 -*-
  2. """
  3. Created on Thu Jun 28 16:17:55 2018

  4. @author: xiaojie
  5. """

  6. import matplotlib.pyplot as plt
  7. import tensorflow as tf   
  8. import numpy as np

  9. image_raw_data = tf.gfile.FastGFile("./datasets/cat.jpg",'rb').read()

  10. with tf.Session() as sess:
  11.     img_data = tf.image.decode_jpeg(image_raw_data)
  12.     print (img_data.get_shape())
  13.     # 输出解码之后的三维矩阵。
  14.     print (img_data.eval())
  15.     img=sess.run(img_data)
  16.     print (img.shape)
  17.     #img_data.set_shape([1797, 2673, 3])
  18.     print (img_data.get_shape())
  19.    

  20. with tf.Session() as sess:         

  21.     boxes = tf.constant([[[0.05, 0.05, 0.9, 0.7], [0.35, 0.47, 0.5, 0.56]]])
  22.     #print ("Digital type: ", img_data.dtype)
  23.     batched = tf.expand_dims(tf.image.convert_image_dtype(img_data, tf.float32), 0)
  24.     #print ("Digital type: ", img_data.dtype)
  25.     image_with_box = tf.image.draw_bounding_boxes(batched, boxes)
  26.    
  27.     #image_with_box_result=tf.image.convert_image_dtype(image_with_box[0], tf.uint8)
  28.     #print ("Digital type: ", image_with_box[0].dtype)
  29.     #print ("Digital type: ", image_with_box_result.dtype)
  30.     plt.imshow(image_with_box.eval().reshape([1797,2673,3])), plt.title('original')
  31.     #plt.subplot(122), plt.imshow(image_with_box_result.eval()), plt.title('result')
  32.     plt.show()
复制代码
有没有人可以帮忙跑一下这个py文件。
总是添加标注框失败。
我知道答案 回答被采纳将会获得10 金币 + 5 金币 已有2人回答
本楼点评(0) 收起

精彩评论2

申克  TF荚荚  发表于 2018-6-28 17:08:01 | 显示全部楼层
程序运行没有问题,但是就是标注框绘制不出来。
本楼点评(0) 收起
neverchange  TF豆豆  发表于 2018-7-4 16:18:01 | 显示全部楼层
本帖最后由 neverchange 于 2018-7-6 00:18 编辑

要用plt.imshow(image_with_box[0].eval().reshape([1797,2673,3])), plt.title('original')即可。
或者plt.imshow(image_with_box[0].eval())也是没问题的。
ps:奇怪,你的帖子为什么不带悬赏金币,默认是必须有的吧
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

7

积分
快速回复 返回顶部 返回列表