74 lines
2.9 KiB
Python
74 lines
2.9 KiB
Python
|
import tensorflow as tf
|
||
|
|
import numpy as np
|
||
|
|
from sklearn.utils import shuffle
|
||
|
|
import matplotlib.pyplot as plot
|
||
|
|
import cv2
|
||
|
|
import vgg
|
||
|
|
|
||
|
|
def read_cifar_file(file, images, labels):
|
||
|
|
shift=0
|
||
|
|
f=np.fromfile(file, dtype=np.uint8)
|
||
|
|
while shift!=f.shape[0]:
|
||
|
|
labels.append(np.eye(10)[f[shift]])
|
||
|
|
shift+=1
|
||
|
|
images.append(f[shift:shift+3*32*32].reshape(3, 32, 32).transpose(1, 2, 0)/255)
|
||
|
|
shift+=3*32*32
|
||
|
|
|
||
|
|
taille_batch=100
|
||
|
|
nbr_entrainement=200
|
||
|
|
labels=['avion', 'automobile', 'oiseau', 'chat', 'cerf', 'chien', 'grenouille', 'cheval', 'bateau', 'camion']
|
||
|
|
|
||
|
|
train_images=[]
|
||
|
|
train_labels=[]
|
||
|
|
read_cifar_file("cifar-10-batches-bin/data_batch_1.bin", train_images, train_labels)
|
||
|
|
read_cifar_file("cifar-10-batches-bin/data_batch_2.bin", train_images, train_labels)
|
||
|
|
read_cifar_file("cifar-10-batches-bin/data_batch_3.bin", train_images, train_labels)
|
||
|
|
read_cifar_file("cifar-10-batches-bin/data_batch_4.bin", train_images, train_labels)
|
||
|
|
read_cifar_file("cifar-10-batches-bin/data_batch_5.bin", train_images, train_labels)
|
||
|
|
|
||
|
|
test_images=[]
|
||
|
|
test_labels=[]
|
||
|
|
read_cifar_file("cifar-10-batches-bin/test_batch.bin", test_images, test_labels)
|
||
|
|
|
||
|
|
images, labels, is_training, sortie, train, accuracy, save=vgg.vggnet(nbr_classes=10, learning_rate=0.01)
|
||
|
|
|
||
|
|
fichier=open("log", "a")
|
||
|
|
with tf.Session() as s:
|
||
|
|
s.run(tf.global_variables_initializer())
|
||
|
|
tab_train=[]
|
||
|
|
tab_test=[]
|
||
|
|
train_images, train_labels=shuffle(train_images, train_labels)
|
||
|
|
for id_entrainement in np.arange(nbr_entrainement):
|
||
|
|
print("> Entrainement", id_entrainement)
|
||
|
|
for batch in np.arange(0, len(train_images), taille_batch):
|
||
|
|
s.run(train, feed_dict={
|
||
|
|
images: train_images[batch:batch+taille_batch],
|
||
|
|
labels: train_labels[batch:batch+taille_batch],
|
||
|
|
is_training: True
|
||
|
|
})
|
||
|
|
print(" entrainement OK")
|
||
|
|
tab_accuracy_train=[]
|
||
|
|
for batch in np.arange(0, len(train_images), taille_batch):
|
||
|
|
p=s.run(accuracy, feed_dict={
|
||
|
|
images: train_images[batch:batch+taille_batch],
|
||
|
|
labels: train_labels[batch:batch+taille_batch],
|
||
|
|
is_training: False
|
||
|
|
})
|
||
|
|
tab_accuracy_train.append(p)
|
||
|
|
print(" train:", np.mean(tab_accuracy_train))
|
||
|
|
tab_accuracy_test=[]
|
||
|
|
for batch in np.arange(0, len(test_images), taille_batch):
|
||
|
|
p=s.run(accuracy, feed_dict={
|
||
|
|
images: test_images[batch:batch+taille_batch],
|
||
|
|
labels: test_labels[batch:batch+taille_batch],
|
||
|
|
is_training: False
|
||
|
|
})
|
||
|
|
tab_accuracy_test.append(p)
|
||
|
|
print(" test :", np.mean(tab_accuracy_test))
|
||
|
|
tab_train.append(1-np.mean(tab_accuracy_train))
|
||
|
|
tab_test.append(1-np.mean(tab_accuracy_test))
|
||
|
|
fichier.write("{:d}:{:f}:{:f}\n".format(id_entrainement, np.mean(tab_accuracy_train), np.mean(tab_accuracy_test)))
|
||
|
|
fichier.close()
|
||
|
|
|
||
|
|
|