def def_params(self):
    self.data_format = 'NHWC'
    if (self.conf.data_type == '3D'):
        self.conv_size = (3, 3, 3)
        self.pool_size = (2, 2, 2)
        (self.axis, self.channel_axis) = ((1, 2, 3), 4)
        self.input_shape = [self.conf.batch, self.conf.depth, self.conf.height, self.conf.width, self.conf.channel]
        self.output_shape = [self.conf.batch, self.conf.depth, self.conf.height, self.conf.width]
    else:
        self.conv_size = (3, 3)
        self.pool_size = (2, 2)
        (self.axis, self.channel_axis) = ((1, 2), 3)
        self.input_shape = [self.conf.batch, self.conf.height, self.conf.width, self.conf.channel]
        self.output_shape = [self.conf.batch, self.conf.height, self.conf.width]
def def_params(self):
    self.data_format = 'NHWC'
    if (self.conf.data_type == '3D'):
        self.conv_size = (3, 3, 3)
        self.pool_size = (2, 2, 2)
        (self.axis, self.channel_axis) = ((1, 2, 3), 4)
        self.input_shape = [self.conf.batch, self.conf.depth, self.conf.height, self.conf.width, self.conf.channel]
        self.output_shape = [self.conf.batch, self.conf.depth, self.conf.height, self.conf.width]
    else:
        self.conv_size = (3, 3)
        self.pool_size = (2, 2)
        (self.axis, self.channel_axis) = ((1, 2), 3)
        self.input_shape = [self.conf.batch, self.conf.height, self.conf.width, self.conf.channel]
        self.output_shape = [self.conf.batch, self.conf.height, self.conf.width]