您的位置:首页 > 大数据 > 人工智能

遇到的问题与解决办法(tf.train.shuffle_batch与tf.train.slice_input_producer)

2017-03-21 14:46 435 查看
1. 在读入数据时,采用了如下代码形式def read_image(image_path):"""the image_path is the path of single image"""file_contents = tf.read_file(input_queue[0])image_data = tf.image.decode_jpeg(file_contents, channels=3)label = input_queue[1]image_id_wgb = input_queue[2]return image_data, label, image_id_wgb
input_queue = tf.train.slice_input_producer([training_image_path_total,training_labels_total, training_image_id_total], shuffle=True)'''the content of input_queue is the properties of one image, which includes a image_path(input_queue[0]), label(input_queue[1]), and image_id(input_queue[2]) of one image'''training_image_data, training_label, training_image_id= read_image(image_path = input_queue)
training_image_batch, training_label_batch,training_image_id_batch = tf.train.shuffle_batch([training_image_data, training_label, training_image_id],batch_size=64,num_threads = 8, min_after_dequeue =101,capacity = 1000)
结果出现以下错误
ValueError: All shapes must be fully defined: [TensorShape([Dimension(None), Dimension(None), Dimension(3)]), TensorShape([]), TensorShape([])]2. 解决办法在 tf.train.shuffle_batch  前加入了下面一个命令,才解决了.原因不知为何training_image_data = tf.image.resize_images(training_image_data, [size, size])
                                            
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: