Stars
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
Image-to-Image Translation in PyTorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
A Deep Learning based project for colorizing and restoring old images (and video!)
Datasets, Transforms and Models specific to Computer Vision
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
End-to-End Object Detection with Transformers
PyTorch implementation of the U-Net for image semantic segmentation with high quality images
Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image …
Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Depth-Aware Video Frame Interpolation (CVPR 2019)
Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also …
Real-Time High-Resolution Background Matting
Learning to See in the Dark. CVPR 2018
Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
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"
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
SwinIR: Image Restoration Using Swin Transformer (official repository)
A Trimap-Free Portrait Matting Solution in Real Time [AAAI 2022]
Tensorflow prebuilt binary for Windows
v2ray linux GUI客户端,支持订阅、vemss、ss等协议,自动更新订阅、检查版本更新
Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
FILM: Frame Interpolation for Large Motion, In ECCV 2022.
[IEEE TMI] Official Implementation for UNet++
micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantization and Training of Neural Networks for Efficient Integer-…