Rui Zhu's starred repositories
google-research
Google Research
vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
nerf-pytorch
A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.
Deformable-Convolution-V2-PyTorch
Deformable ConvNets V2 (DCNv2) in PyTorch
manhattan_sdf
Code for "Neural 3D Scene Reconstruction with the Manhattan-world Assumption" CVPR 2022 Oral
perceiver-io
A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training
Total3DUnderstanding
Implementation of CVPR'20 Oral: Total3DUnderstanding: Joint Layout, Object Pose and Mesh Reconstruction for Indoor Scenes from a Single Image
ssn_superpixels
Superpixel Sampling Networks (ECCV2018)
docker-nvidia-egl-desktop
KDE Plasma Desktop container designed for Kubernetes with direct access to the GPU with EGL using VirtualGL and Vulkan with WebRTC and HTML5, providing an open-source remote cloud graphics or game streaming platform. Does not require /tmp/.X11-unix host sockets or host configuration.
panoptic-reconstruction
Official implementation of the NeurIPS 2021 paper "Panoptic 3D Scene Reconstruction from a Single RGB Image"
chamfer_distance
Implementation of the Chamfer Distance as a module for PyTorch
FocusOnDepth
A Monocular depth-estimation for in-the-wild AutoFocus application.
gist-alfred
An alfred workflow for accessing gists
cooperative_scene_parsing
Code for NeurIPS 2018: Cooperative Holisctic Scene Understanding: Unifying 3D Object, Layout, and Camera Pose Estimation
OptixRenderer
An optix-based GPU path tracer
lighthouse
code release for CVPR 2020 Paper Lighthouse: Predicting Lighting Volumes for Spatially-Coherent Illumination https://people.eecs.berkeley.edu/~pratul/lighthouse/
deformable_conv2d_pytorch
deformable_conv2d layer implemented in pytorch
PyTorch-extension-Convolution
An example of C++ extension for PyTorch.