places365.py 文件源码

def decoderFn(num_samples=1):
  class decoder_func(slim.data_decoder.DataDecoder):
    @staticmethod
    def list_items():
      return ['image', 'label']


    @staticmethod
    def decode(data, items):
      image_buffer = _decode_from_string(data)
      # if num_samples == 1:
        # tf.Assert(tf.shape(image_buffer)[0] == 1, image_buffer)
        # image_buffer = image_buffer[0]
      # else:
      image_buffer = tf.pack(image_buffer)
      return image_buffer
  return decoder_func