Initial commit
This commit is contained in:
98
Tensorflow/tutoriel10/CIFAR_10_vgg.py
Normal file
98
Tensorflow/tutoriel10/CIFAR_10_vgg.py
Normal file
@@ -0,0 +1,98 @@
|
||||
import tensorflow as tf
|
||||
import numpy as np
|
||||
import matplotlib.pyplot as plot
|
||||
import cv2
|
||||
import vgg
|
||||
from sklearn.utils import shuffle
|
||||
|
||||
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=50
|
||||
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)
|
||||
|
||||
def transform_img(img):
|
||||
img=tf.image.random_flip_left_right(img)
|
||||
img=tf.image.random_hue(img, 0.08)
|
||||
img=tf.image.random_saturation(img, 0.6, 1.6)
|
||||
img=tf.image.random_brightness(img, 0.05)
|
||||
img=tf.image.random_contrast(img, 0.7, 1.3)
|
||||
x=int(img.shape[0])
|
||||
y=int(img.shape[1])
|
||||
z=int(img.shape[2])
|
||||
img=tf.image.random_crop(img, [int(x*0.90), int(y*0.90), z])
|
||||
img=tf.image.resize_images(img, (x, y))
|
||||
return(img)
|
||||
|
||||
fichier=open("log", "a")
|
||||
with tf.Session() as s:
|
||||
s.run(tf.global_variables_initializer())
|
||||
tab_train=[]
|
||||
tab_test=[]
|
||||
|
||||
train_images=np.array(train_images, dtype=np.float32)
|
||||
train_images2=tf.map_fn(transform_img, train_images)
|
||||
train_images3=tf.map_fn(transform_img, train_images)
|
||||
train_images4=tf.map_fn(transform_img, train_images)
|
||||
train_images=tf.concat([train_images, train_images2, train_images3, train_images4], axis=0)
|
||||
train_labels=np.array(train_labels)
|
||||
train_labels=tf.concat([train_labels, train_labels, train_labels, train_labels], axis=0)
|
||||
|
||||
train_images=s.run(train_images)
|
||||
train_labels=s.run(train_labels)
|
||||
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()
|
||||
|
||||
|
||||
BIN
Tensorflow/tutoriel10/Figure_1.png
Normal file
BIN
Tensorflow/tutoriel10/Figure_1.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 84 KiB |
15
Tensorflow/tutoriel10/README.md
Normal file
15
Tensorflow/tutoriel10/README.md
Normal file
@@ -0,0 +1,15 @@
|
||||
# Tutoriel tensorflow
|
||||
## Surapprentissage: complétion du dataset
|
||||
|
||||
La vidéo de ce tutoriel est disponible à l'adresse suivante: https://www.youtube.com/watch?v=rLSnx0LiObo
|
||||
|
||||
## CIFAR10
|
||||
|
||||
N'oubliez pas de récuperer la base cifar10 (binary version) à l'adresse suivante:
|
||||
https://www.cs.toronto.edu/~kriz/cifar.html
|
||||
|
||||
### Courbes d'erreur sur la base de validation:
|
||||

|
||||
|
||||
L'apprentissage prend 2h50 sur une GeForce 1080
|
||||
|
||||
102
Tensorflow/tutoriel10/vgg.py
Normal file
102
Tensorflow/tutoriel10/vgg.py
Normal file
@@ -0,0 +1,102 @@
|
||||
import tensorflow as tf
|
||||
import numpy as np
|
||||
|
||||
def convolution(couche_prec, taille_noyau, nbr_noyau):
|
||||
w=tf.Variable(tf.random.truncated_normal(shape=(taille_noyau, taille_noyau, int(couche_prec.get_shape()[-1]), nbr_noyau)))
|
||||
b=np.zeros(nbr_noyau)
|
||||
result=tf.nn.conv2d(couche_prec, w, strides=[1, 1, 1, 1], padding='SAME')+b
|
||||
return result
|
||||
|
||||
def fc(couche_prec, nbr_neurone):
|
||||
w=tf.Variable(tf.random.truncated_normal(shape=(int(couche_prec.get_shape()[-1]), nbr_neurone), dtype=tf.float32))
|
||||
b=tf.Variable(np.zeros(shape=(nbr_neurone)), dtype=tf.float32)
|
||||
result=tf.matmul(couche_prec, w)+b
|
||||
return result
|
||||
|
||||
def vggnet(nbr_classes, learning_rate=1E-3, momentum=0.99):
|
||||
ph_images=tf.placeholder(shape=(None, 32, 32, 3), dtype=tf.float32, name='images')
|
||||
ph_labels=tf.placeholder(shape=(None, nbr_classes), dtype=tf.float32)
|
||||
ph_is_training=tf.placeholder_with_default(False, (), name='is_training')
|
||||
|
||||
result=convolution(ph_images, 3, 64)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.layers.dropout(result, 0.2, training=ph_is_training)
|
||||
result=tf.nn.relu(result)
|
||||
result=convolution(result, 3, 64)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.2, training=ph_is_training)
|
||||
result=tf.nn.max_pool(result, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME')
|
||||
|
||||
result=convolution(result, 3, 128)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.2, training=ph_is_training)
|
||||
result=convolution(result, 3, 128)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.3, training=ph_is_training)
|
||||
result=tf.nn.max_pool(result, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME')
|
||||
|
||||
result=convolution(result, 3, 256)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.3, training=ph_is_training)
|
||||
result=convolution(result, 3, 256)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.3, training=ph_is_training)
|
||||
result=convolution(result, 3, 256)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.3, training=ph_is_training)
|
||||
result=tf.nn.max_pool(result, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME')
|
||||
|
||||
result=convolution(result, 3, 512)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.3, training=ph_is_training)
|
||||
result=convolution(result, 3, 512)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.4, training=ph_is_training)
|
||||
result=convolution(result, 3, 512)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.4, training=ph_is_training)
|
||||
result=tf.nn.max_pool(result, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME')
|
||||
|
||||
result=convolution(result, 3, 512)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.4, training=ph_is_training)
|
||||
result=convolution(result, 3, 512)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.4, training=ph_is_training)
|
||||
result=convolution(result, 3, 512)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.nn.relu(result)
|
||||
result=tf.layers.dropout(result, 0.5, training=ph_is_training)
|
||||
result=tf.nn.max_pool(result, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME')
|
||||
result=tf.contrib.layers.flatten(result)
|
||||
|
||||
result=fc(result, 1024)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.layers.dropout(result, 0.5, training=ph_is_training)
|
||||
result=tf.nn.relu(result)
|
||||
result=fc(result, 1024)
|
||||
result=tf.layers.batch_normalization(result, training=ph_is_training, momentum=momentum)
|
||||
result=tf.layers.dropout(result, 0.5, training=ph_is_training)
|
||||
result=tf.nn.relu(result)
|
||||
result=fc(result, nbr_classes)
|
||||
socs=tf.nn.softmax(result, name="sortie")
|
||||
|
||||
loss=tf.nn.softmax_cross_entropy_with_logits_v2(labels=ph_labels, logits=result)
|
||||
extra_update_ops=tf.get_collection(tf.GraphKeys.UPDATE_OPS)
|
||||
with tf.control_dependencies(extra_update_ops):
|
||||
train=tf.train.AdamOptimizer(learning_rate).minimize(loss)
|
||||
accuracy=tf.reduce_mean(tf.cast(tf.equal(tf.argmax(socs, 1), tf.argmax(ph_labels, 1)), tf.float32))
|
||||
|
||||
return ph_images, ph_labels, ph_is_training, socs, train, accuracy, tf.train.Saver()
|
||||
|
||||
Reference in New Issue
Block a user