CosmiQ_SN4_Baseline
CosmiQ_SN4_Baseline copied to clipboard
Bug found in DataGenerator.py
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.