File size: 798 Bytes
5fee096 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | import numpy as np
class LinearBuffer:
def __init__(self, buffer_size, strategy, batch_size):
self.buffer_size = buffer_size
self.strategy = strategy
self.batch_size = batch_size
self.total_classes = 0
self.images, self.labels = [], []
def is_empty(self):
return len(self.labels) == 0
class LinearSpiltBuffer:
def __init__(self, buffer_size, strategy, batch_size, val_ratio):
self.buffer_size = buffer_size
self.strategy = strategy
self.batch_size = batch_size
self.val_ratio = 0.1
self.total_classes = 0
self.train_images, self.train_labels = [], []
self.val_images, self.val_labels = [], []
def is_empty(self):
return len(self.train_labels) == 0 |