Question

我正在尝试修改serving tutorial以使用我的模型，这基本上是修改为使用CSV文件和JPEG的CIFAR示例。我似乎无法找到Exporter类的文档，但这是我到目前为止所拥有的。它位于cifar10_train.py文件中的train（）函数中：

  labels = numpy.fromfile(os.path.join(data_dir, 'labels.txt'), dtype=numpy.int32, count=-1, sep='\n')

  filenames_and_labels = []

  start_image_number = 1
  end_image_number = 8200

  for i in xrange(start_image_number, end_image_number):
    file_name = os.path.join(data_dir, 'image%d.jpg' % i)
    label = labels[i - 1]
    filenames_and_labels.append(file_name + "," + str(label))


  print('Reading filenames for ' + str(len(filenames_and_labels)) + ' files (from ' + str(start_image_number) + ' to ' + str(end_image_number) + ')')

  for filename_and_label in filenames_and_labels:
    array = filename_and_label.split(",")
    f = array[0]
    # print(array)
    if not tf.gfile.Exists(f):
      raise ValueError('Failed to find file: ' + f)

  # Create a queue that produces the filenames to read.
  filename_and_label_queue = tf.train.string_input_producer(filenames_and_labels)

  filename_and_label_tensor = filename_and_label_queue.dequeue()
  filename, label = tf.decode_csv(filename_and_label_tensor, [[""], [""]], ",")
  file_contents = tf.read_file(filename)
  image = tf.image.decode_jpeg(file_contents)

以下是我用来训练模型的代码：

{{1}}

我是如何正确设置出口商的？

Answer 1

请查看MNIST export example。

显示如何生成x和y然后放入签名中。

此外，Inception example显示了如何扩展现有模型以创建导出和服务。特别是cifar10.inference调用类似于inception_model.inference。

出口商分类_签名

1 个答案: