AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-04 12:10:45 +02:00
Code Issues Projects Releases Packages Wiki Activity

Suite de test brutus

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-11-20 16:06:27 -05:00
parent cc737b7997
commit 0e7ec8b5b0
6 changed files with 65 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

4
PBA/LeNet.py
View file

@ -31,7 +31,7 @@ def LeNet(images, num_classes):
n_n_fc2 = 500; # number of neurons of first fully connected layer (default = 576)
# 1.layer: convolution + max pooling
W_conv1_tf = weight_variable([s_f_conv1, s_f_conv1, 1, n_f_conv1], name = 'W_conv1_tf') # (5,5,1,32)
W_conv1_tf = weight_variable([s_f_conv1, s_f_conv1, images.shape[3], n_f_conv1], name = 'W_conv1_tf') # (5,5,1,32)
b_conv1_tf = bias_variable([n_f_conv1], name = 'b_conv1_tf') # (32)
h_conv1_tf = tf.nn.relu(conv2d(images,
W_conv1_tf) + b_conv1_tf,
@ -76,4 +76,4 @@ def LeNet(images, num_classes):
# tf.argmax(y_data_tf, 1),
# name = 'y_pred_correct_tf')
logits = z_pred_tf
return logits #y_pred_proba_tf
return logits #y_pred_proba_tf