Initialize neural network
Code for collecting image samples