Skip to content
Snippets Groups Projects

Gan

Closed Guillaume HEUSCH requested to merge gan into master
1 file
+ 9
1
Compare changes
  • Side-by-side
  • Inline
@@ -79,9 +79,17 @@ class Memory(Base):
@@ -79,9 +79,17 @@ class Memory(Base):
indexes = numpy.array(range(self.data.shape[0]))
indexes = numpy.array(range(self.data.shape[0]))
numpy.random.shuffle(indexes)
numpy.random.shuffle(indexes)
selected_data = self.data[indexes[0:self.batch_size], ...]
selected_data = numpy.zeros(shape=self.shape)
 
for i in range(self.batch_size):
 
 
selected_data[i, ...] = self.data[indexes[i], ...]
 
# Scaling
 
selected_data[i, ...] = self.normalize_sample(selected_data[i, ...])
 
selected_labels = self.labels[indexes[0:self.batch_size]]
selected_labels = self.labels[indexes[0:self.batch_size]]
 
 
# Applying the data augmentation
# Applying the data augmentation
if self.data_augmentation is not None:
if self.data_augmentation is not None:
for i in range(selected_data.shape[0]):
for i in range(selected_data.shape[0]):
Loading