Ver código fonte

Merge branch 'mnist'

DricomDragon 5 anos atrás
pai
commit
70d36c5af5
2 arquivos alterados com 35 adições e 1 exclusões
  1. 12 1
      README.md
  2. 23 0
      python/src/io_mnist.py

+ 12 - 1
README.md

@@ -1,3 +1,14 @@
 # NeuralNetwork
 
-Practice neural network with number recognition.
+Practice neural network with number recognition.
+
+## Requirements
+
+### Download data samples
+
+Execute the script `resources/retrieve_digit_database.bash`.
+
+### Install python dependencies
+
+`pip3 install python-mnist`
+

+ 23 - 0
python/src/io_mnist.py

@@ -0,0 +1,23 @@
+from mnist import MNIST
+import numpy as np
+
+def load_training_samples():
+	"""
+	Return np_images, labels
+	where
+	np_impages is a np.array of 60 000 x 784
+	labels is a python list of expected answers
+	"""
+	mndata = MNIST('../../resources/download')
+
+	images = [[]] # Contains vectors of 784 pixels image
+	labels = [] # Contains expected response for each image
+
+	images, labels = mndata.load_training()
+
+	np_images = np.array(images, dtype=np.float64)
+
+	# Normalize data between 0.0 and 1.0
+	np_images /= 255
+
+	return np_images, labels