-
Carnegie Mellon University
Stars
General Vision Benchmark, GV-B, a project from OpenGVLab
CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped, CVPR 2022
OpenDILab RL HPC OP Lib, including CUDA and Triton kernel
OpenDILab RL Kubernetes Custom Resource and Operator Lib
Decision Intelligence Platform for Autonomous Driving simulation.
An artificial intelligence platform for the StarCraft II with large-scale distributed training and grand-master agents.
OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.
PyTorch package for the discrete VAE used for DALL·E.
Official Implementation for "StyleCLIP: Text-Driven Manipulation of StyleGAN Imagery" (ICCV 2021 Oral)
NVIDIA's Deep Imagination Team's PyTorch Library
Count the MACs / FLOPs of your PyTorch model.
[ ECCV 2020 Spotlight ] Pytorch implementation for "Distribution-Balanced Loss for Multi-Label Classification in Long-Tailed Datasets"
Simplest working implementation of Stylegan2, state of the art generative adversarial network, in Pytorch. Enabling everyone to experience disentanglement
[CVPR 2021] Actor-Context-Actor Relation Network for Spatio-temporal Action Localization
You Only Watch Once: A Unified CNN Architecture for Real-Time Spatiotemporal Action Localization
An efficient video loader for deep learning with smart shuffling that's super easy to digest
[ICML'20] Multi Steepest Descent (MSD) for robustness against the union of multiple perturbation models.
Code of the STAGE module for video action detection
PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.
Repository containing Pytorch code for EKFAC and K-FAC perconditioners.
[ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding
PyTorch implementation of Multi-Label Image Recognition with Graph Convolutional Networks, CVPR 2019.