Visualization of neural network architectures and parameters in VR.
This tool can be used to render a processed neural network in VR.
- The tool uses pyopenvr for now`and needs SteamVR installed to work
requirements_vr.txt
pip install -r requirements_vr.txt
python start_tool_vr.py
Or
- Run
start_tool.py --demo
to download data of an already processed model and render it.
Using Oculus Quest 2 controller:
- left/right trigger: shrink/grow model
- left/right grip: attach model to left/right hand
- left/right joystick: rotate model when grabbed
- pressing left/right joystick: reset model position and orientation
- x/b button: rotate between some predefined rendering modes for nodes and edges
- y/a button: rotate between different class highlighting settings
See README.md for information on the desktop GUI
- Windows 10
- NVIDIA GeForce RTX 3080
- AMD Ryzen 7 3700X
- Oculus Quest 2