Ejemplo n.º 1
0
from alex_net import AlexNet
import tensorflow as tf
import numpy as np
import pic_to_csv
'''
train=np.random.normal(size=(100,224,224,3))
print(train.shape)

labels=np.random.random(size=100)*4
labels=labels.astype(np.int)
'''
x, labels = pic_to_csv.getDataSet()
print(x.shape)
print(labels.shape)

model = AlexNet()
model.fit(x, labels, epochs=10, batch_size=5)
Ejemplo n.º 2
0
from alex_net import AlexNet
import tensorflow as tf
import numpy as np

train = np.random.normal(size=(100, 224, 224, 3))
print(train.shape)

labels = np.random.random(size=100) * 4
labels = labels.astype(np.int)

model = AlexNet()
model.fit(train, labels, epochs=10, batch_size=10)