This is a Dx12 application which runs a neural network on the GPU using HLSL shaders, intended as a resource to learn about deep learning and neural networks. The network is trained to reproduce an input image (it learns a mapping from UV coordinates to texel values). Most interesting implementation details are found in a single HLSL file - Dx12NN.hlsl.
Application shows a reference image (training image that it's trying to learn) on the left, and prediction on the right.
For the first build, run the Visual Studio as an Administrator. Build process will create symbolic links to folder with shaders, which can fail on some systems if VS is run without administrator rights.
- Run
cmakeon the src subfolder and build the generated project - Or run provided file
create_vs_project.batand build the project generated in the vsbuild subfolder (requires Visual Studio 2022)
Note: First start takes about a minute, because shaders need to be compiled. Subsequent runs are faster.
If you get an error Error: failed to get shader blob result!, it most likely means that build didn't create a symbolic link for shaders folder in the build directory and executable can't find shader files. To fix this, run Visual Studio as administrator, clean the solution and run the build again.
This project uses following dependencies:
- d3dx12.h, provided with an MIT license.
- STB library, provided with an MIT license.
- DXC Compiler, provided with an University of Illinois Open Source
- ImGUI, provided with an MIT license.
- GLM library, provided with an MIT license.