(self, image_shape, is_train)
return py_reader
build_input
build_input
build_input
def build_input(self, image_shape, is_train):
        """
            build_input
        """
        name = 'train_reader' if is_train else 'test_reader'
        py_reader = fluid.layers.py_reader(
            capacity=64,
            shapes=[[-1] + image_shape, [-1, 1]],
            lod_levels=[0, 0],
            dtypes=["float32", "int64"],
            use_double_buffer=True,
            name=name)
        return py_reader
def build_input ( self , image_shape , is_train ) : name = 'train_reader' if is_train else 'test_reader' py_reader = fluid . layers . py_reader ( capacity = 64 , shapes = [ [ - 1 ] + image_shape , [ - 1 , 1 ] ] , lod_levels = [ 0 , 0 ] , dtypes = [ "float32" , "int64" ] , use_double_buffer = True , name = name ) return py_reader