search

humandotlearning / hand-written-numbers-recognition-application

application using neural network and MNIST dataset
2 stars 1 forks source link

Testing mnist_read_numbers.ipynb #1

Closed salvadorrueda closed 5 years ago

salvadorrueda commented 5 years ago

I'm trying to test mnist_read_numbers.ipynb Jupyter notebook with no success

First I tried in colab (google) and I'm still fixing issues like install pytorch !pip install -q ... choose GPU intead CPU upload the model file 'model_gpu.pt' ... and it doesn't work yet.

I think the problem is try to use openCV VideoCapture on google colab.

It would be useful write some kind of instructions in to order to be able to execute the mnist_read_numbers.ipynb like you did in the youtube video.

Thanks.

salvadorrueda commented 5 years ago

Finally works 😄

You need a computer with a webcam and install Jupyter Notebook -> https://www.anaconda.com/download pythorch -> conda install pytorch torchvision -c pytorch OpenCV -> conda install opencv

Note that: # MacOS Binaries dont support CUDA, install from source if CUDA is needed

You can use cpu instead gpu (CUDA) by changing it in the code.