Stars
PyTorch code and models for the DINOv2 self-supervised learning method.
A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis
Read, write and manipulate GP3, GP4 and GP5 files.
A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
Computational Pathology Toolbox developed by TIA Centre, University of Warwick.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
ACORN: Adaptive Coordinate Networks for Neural Scene Representation | SIGGRAPH 2021
Computes vesselness scores for 3-dimensional images.
Mean Absolute Error Does Not Treat Examples Equally and Gradient Magnitude’s Variance Matters
Official Release of ICLR 2020 paper "SCALOR: Generative World Models with Scalable Object Representations"
Tool for reading and writing datasets of tensors in a Lightning Memory-Mapped Database (LMDB). Designed to manage machine learning datasets with fast reading speeds.
A Toolbox for Adversarial Robustness Research
Why ReLU networks yield high-confidence predictions far away from the training data and how to mitigate the problem [CVPR 2019, oral]
Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch
Convolutional Neural Network for 3D meshes in PyTorch
The Pytorch implementation of the 3D Anisotropic Hybrid Network described in the paper "3D Anisotropic Hybrid Network: Transferring Convolutional Features from 2D Images to 3D Anisotropic Volumes"
Pure Numpy Implementation of the Coherent Point Drift Algorithm
The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on d…
Graph Neural Network Library for PyTorch
An optimizer that trains as fast as Adam and as good as SGD.
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
Differentiable architecture search for convolutional and recurrent networks
BiBLaTeX style for Springer Lecture Notes in Computer Science
I decide to sync up this repo and self-critical.pytorch. (The old master is in old master branch for archive)