cakesibyl

TensorRT

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.

yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

alpa

Training and serving large-scale neural networks with auto parallelization.

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Real-time-GesRec

Real-time Hand Gesture Recognition with PyTorch on EgoGesture, NvGesture, Jester, Kinetics and UCF101

sense

Enhance your application with the ability to see and interact with humans using any RGB camera.

pybind11_examples

Examples for the usage of "pybind11"

srt

Secure, Reliable, Transport

onnx

Open standard for machine learning interoperability

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

mediapipe

Cross-platform, customizable ML solutions for live and streaming media.

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), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

DiverseBranchBlock

Diverse Branch Block: Building a Convolution as an Inception-like Unit

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

MobileNetV3

An implementation of MobileNetV3 with pyTorch

torchnvjpeg

Decode JPEG image on GPU using PyTorch

KAIR

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

RFDNet-ImageSuperResolution

Keras Implementation of the paper Residual Feature Distillation Network for Lightweight Image Super-Resolution

jetson-inference

Hello AI World guide to deploying deep-learning inference networks and deep vision primitives with TensorRT and NVIDIA Jetson.

RFDN

HR-NAS

HR-NAS: Searching Efficient High-Resolution Neural Architectures with Lightweight Transformers (CVPR21 Oral)

VRT

VRT: A Video Restoration Transformer (official repository)

TDAN-VSR-CVPR-2020

TDAN: Temporally-Deformable Alignment Network for Video Super-Resolution, CVPR 2020

Zooming-Slow-Mo-CVPR-2020

Fast and Accurate One-Stage Space-Time Video Super-Resolution (accepted in CVPR 2020)

IMDN

detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

hyperseg

HyperSeg - Official PyTorch Implementation

STDC-Seg

Source Code of our CVPR2021 paper "Rethinking BiSeNet For Real-time Semantic Segmentation"