hmorimitsu

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

lightning

Deep learning framework to train, deploy, and ship AI products Lightning fast.

nvtop

GPU & Accelerator process monitoring for AMD, Apple, Huawei, Intel, NVIDIA and Qualcomm

taming-transformers

Taming Transformers for High-Resolution Image Synthesis

arxiv-sanity-preserver

Web interface for browsing, search and filtering recent arxiv submissions

RAFT

LoFTR

Code for "LoFTR: Detector-Free Local Feature Matching with Transformers", CVPR 2021, T-PAMI 2022

sam

SAM: Sharpness-Aware Minimization (PyTorch)

deep-learning-project-template

Pytorch Lightning code guideline for conferences

landmark-detection

Four landmark detection algorithms, implemented in PyTorch.

DenseMatching

Dense matching library based on PyTorch

PatchmatchNet

Official code of PatchmatchNet (CVPR 2021 Oral)

SelFlow

SelFlow: Self-Supervised Learning of Optical Flow

MaskFlownet

[CVPR 2020, Oral] MaskFlownet: Asymmetric Feature Matching with Learnable Occlusion Mask

GMA

Learning to Estimate Hidden Motions with Global Motion Aggregation (ICCV 2021)

LiteFlowNet2

A Lightweight Optical Flow CNN - Revisiting Data Fidelity and Regularization, TPAMI 2020

FastFlowNet

FastFlowNet: A Lightweight Network for Fast Optical Flow Estimation (ICRA 2021)

LiteFlowNet3

LiteFlowNet3: Resolving Correspondence Ambiguity for More Accurate Optical Flow Estimation, ECCV 2020

GLU-Net

Official implementation of the paper GLU-Net (CVPR2020-Oral)

spynet

Spatial Pyramid Network for Optical Flow

DGC-Net

A PyTorch implementation of "DGC-Net: Dense Geometric Correspondence Network"

SCV

Learning Optical Flow from a Few Matches (CVPR 2021)

R2D2

DICL-Flow

[NeurIPS 2020] Displacement-Invariant Matching Cost Learning for Accurate Optical Flow Estimation

ScopeFlow

Dynamic Scene Scoping for Optical Flow (CVPR 2020)

MaskFlownet-Pytorch

Pytorch implementation of MaskFlownet

GOCor

Official Pytorch implementation of 'GOCor: Bringing Globally Optimized Correspondence Volumes into Your Neural Network' (NeurIPS 2020)

ppac_refinement

Probabilistic Pixel-Adaptive Refinement Networks (CVPR 2020)

RAFT-NCUP

The official PyTorch implementation for "Normalized Convolution Upsampling for Refined Optical Flow Estimation"

LCV

Learnable Cost Volume Using the Cayley Representation