Guest User

Untitled

a guest
Nov 17th, 2017
384
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.01 KB | None | 0 0
  1. ValueError: Cannot feed value of shape (48, 1) for Tensor 'TargetsData/Y:0', which has shape '(?, 2)'
  2.  
  3. import deepneuralnet as net
  4. import numpy as np
  5. from tflearn.data_utils import image_preloader
  6. import os
  7.  
  8. model = net.model
  9. train_path = os.path.abspath('train')
  10. print(train_path)
  11. X, Y = image_preloader(target_path=train_path, image_shape=(100, 100),
  12. mode='folder', grayscale=False, categorical_labels=True, normalize=True)
  13. X = np.reshape(X, (-1, 100, 100, 3))
  14.  
  15. validate_path = os.path.abspath('validate')
  16. W, Z = image_preloader(target_path=validate_path, image_shape=(100, 100),
  17. mode='folder', grayscale=False, categorical_labels=True, normalize=True)
  18. W = np.reshape(W, (-1, 100, 100, 3))
  19. model.fit(X, Y, n_epoch=250, validation_set=(W,Z), show_metric=True)
  20. model.save('./ZtrainedNet/final-model.tfl')
  21.  
  22. import tflearn
  23. from tflearn.layers.core import input_data, dropout, fully_connected
  24. from tflearn.layers.conv import conv_2d, max_pool_2d
  25. from tflearn.layers.estimator import regression
  26. from tflearn.metrics import Accuracy
  27.  
  28. acc = Accuracy()
  29. network = input_data(shape=[None, 100, 100, 3])
  30. # Conv layers ------------------------------------
  31. network = conv_2d(network, 64, 3, strides=1, activation='relu')
  32. network = max_pool_2d(network, 2, strides=2)
  33. network = conv_2d(network, 64, 3, strides=1, activation='relu')
  34. network = max_pool_2d(network, 2, strides=2)
  35. network = conv_2d(network, 64, 3, strides=1, activation='relu')
  36. network = conv_2d(network, 64, 3, strides=1, activation='relu')
  37. network = conv_2d(network, 64, 3, strides=1, activation='relu')
  38. network = max_pool_2d(network, 2, strides=2)
  39. # Fully Connected Layers -------------------------
  40. network = fully_connected(network, 1024, activation='tanh')
  41. network = dropout(network, 0.5)
  42. network = fully_connected(network, 1024, activation='tanh')
  43. network = dropout(network, 0.5)
  44. network = fully_connected(network, 2, activation='softmax')
  45. network = regression(network, optimizer='momentum',
  46. loss='categorical_crossentropy',
  47. learning_rate=0.001, metric=acc)
  48. model = tflearn.DNN(network)
Add Comment
Please, Sign In to add comment