CNN accelerated by cuda. Test on mnist and finilly get 99.7%
- Use DropConnnect to train the NetWork
- Support checkpoint, the program will save the best test result and save the network weight in the file "net.txt"
- Translate the data set of mnist, including scale, rotate, distortion.
Depend on opencv and cuda You can compile the code on windows or linux.
###Windows
- Install vs2010.
- Download and install opencv-2.4 or other higher versions
- Download and install <a href="https://developer.nvidia.com/cuda-downloads", title="cuda-5.0"> cuda-5.0 or other higher versions
- When you create a new project using VS2010, You can find NVIDIA-CUDA project template, create a cuda-project.
- Add the opencv "include path" and "lib path" to the project
- Our project use cublas.lib and curand.lib from cuda, so the project's link list should include these libs
###Linux
- Install opencv and cuda
- Start the nsight from cuda
- Add the opencv "include path" and "lib path" to the project
- Link list should include cublas.lib and curand.lib
Author :zhxfl
mail :zhxfl@mail.ustc.edu.cn
Welcome for any suggest!!