CosmiQ_SN4_Baseline icon indicating copy to clipboard operation
CosmiQ_SN4_Baseline copied to clipboard

Bug found in DataGenerator.py

Open Zeltserj opened this issue 4 years ago • 0 comments

Hi, In all the generator classes, the length function seems to be implemented wrong. The implementation:

    def __len__(self):
        'Denotes the number of batches per epoch'
        return int(np.floor(self.images.shape[1]/self.batch_size))

when actually the nominator should be self.image.shape[0], which is number of sample and not the height of the input image.

Zeltserj avatar Jun 11 '20 07:06 Zeltserj