generators.py 文件源码

def __init__(self, filter_size, input_shape, filters_in, batch_size):
        """
        Parameters:
        -----------
        filter_size : int
            Size of the filter in 1 dimension (total = filter_size ** 2)
        input_shape: list_like
            Size of input image this filter is working on. This is used for
            generating separate filters for each pixel position of the image
        filter_in : int
            Number of channels in input
        batch_size : int
            Batch size
        """

        self.filter_size = filter_size
        self.input_shape = input_shape
        self.filters_in = filters_in
        self.batch_size = batch_size
        self.init = initializations.get("glorot_uniform")
        self.b_init = initializations.get("zero")

        self.setup_weights()