Amir-Arsalan

U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

MiDaS

Code for robust monocular depth estimation described in "Ranftl et. al., Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfer, TPAMI 2022"

SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

contrastive-unpaired-translation

Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)

neuraloperator

Learning in infinite dimension with neural operators.

siren

Official implementation of "Implicit Neural Representations with Periodic Activation Functions"

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

gradslam

gradslam is an open source differentiable dense SLAM library for PyTorch

nvdiffrast

Nvdiffrast - Modular Primitives for High-Performance Differentiable Rendering

NVAE

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)

clean-fid

PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]

torchsort

Fast, differentiable sorting and ranking in PyTorch

blender-cli-rendering

Blender Python scripts for rendering images directly from command-line interface

symbolic_deep_learning

Code for "Discovering Symbolic Models from Deep Learning with Inductive Biases"

dna-3d-engine

3d engine implementation in DNA code!

siren

PyTorch implementation of Sinusodial Representation networks (SIREN)

GrabNet

GrabNet: A Generative model to generate realistic 3D hands grasping unseen objects (ECCV2020)

plan2explore

Repository for the paper "Planning to Explore via Self-Supervised World Models"

Neural-Localization

Train an RL agent to localize actively (PyTorch)

hessian_penalty

Official Implementation of "The Hessian Penalty: A Weak Prior for Unsupervised Disentanglement" (ECCV 2020 Spotlight)

latent-composition

Generative-3D-Part-Assembly

Generative 3D Part Assembly via Dynamic Graph Learning, NeurIPS 2020

psdr-cuda

Path-space differentiable renderer

focal_calibration

Code for the paper "Calibrating Deep Neural Networks using Focal Loss"

torch_tem

Implementation of the Tolman Eichenbaum Machine in pytorch

Regularized_autoencoders-RAE-

procedural_objects

blender based random procedural object generation for bullet grasping

moog.github.io

Modular Object-Oriented Games (MOOG): Python-based game engine for reinforcement learning, psychology, and neurophysiology.

unicornn

Official code for UnICORNN (ICML 2021)

strawberry-rats