I run into a TypeError
when I run my code on Floydhub. I can successfully run my code on Kaggle and Colab, but when I try running it here it error:
TypeErrorTraceback (most recent call last)
<ipython-input-11-e5a1940f2be6> in <module>()
7 # create train and valid datagens
8 train_generator = traingenerator.create_train(
----> 9 train_dataset_info[train_indexes], 32, (SIZE,SIZE,3), augument=True)
10 validation_generator = data_generator.create_train(
11 train_dataset_info[valid_indexes], 32, (SIZE,SIZE,3), augument=False)
TypeError: unbound method create_train() must be called with traingenerator instance as first argument (got ndarray
instance instead)
For some reason it's not calling my class function:
class traingenerator:
def create_train(dataset_info, batch_size, shape, augument=True):
assert shape[2] == 3
while True:
dataset_info = shuffle(dataset_info)
for start in range(0, len(dataset_info), batch_size):
end = min(start + batch_size, len(dataset_info))
batch_images = []
X_train_batch = dataset_info[start:end]
batch_labels = np.zeros((len(X_train_batch), 28))
for i in range(len(X_train_batch)):
image = data_generator.load_image(
X_train_batch[i]['path'], shape)
if augument:
image = data_generator.augment(image)
batch_images.append(image/255.)
batch_labels[i][X_train_batch[i]['labels']] = 1
yield np.array(batch_images, np.float32), batch_labels