Skip to content
@megvii-research

MEGVII Research

Power Human with AI. 持续创新拓展认知边界 非凡科技成就产品价值
README.md

Megvii Research

Continuous Innovation Expands Horizons and Broadens the Mind Leveraging Cutting-Edge Technologies to Create Tangible Value

homepage | bilibili | 知乎 | 中文主页

The following list also includes projects from some of Megvii Research's affiliated organizations:

Foundation Model | Base Detection | 3D team | MegEngine

name description
4K-Face Image 4K-Face: A Dataset with Huge Scale-variance Faces
AAAI2023-PVD Image Official Implementation of PVD:One is All: Bridging the Gap Between Neural Radiance Fields Architectures with Progressive Volume Distillation
ACON Image None
AGFlow Image Learning Optical Flow with Adaptive Graph Reasoning (AGFlow, AAAI-2022)
AnchorDETR Image An official implementation of the Anchor DETR.
AngleNAS Image None
Arch-Net Image Arch-Net: Model Distillation for Architecture Agnostic Model Deployment
AutoAssign Image Pytorch implementation of "AutoAssign: Differentiable Label Assignment for Dense Object Detection"
BBN Image The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition
BEVDepth Image Official code for BEVDepth.
BEVStereo Image Official code for BEVStereo
BasesHomo Image The official PyTorch implementation of the paper "Motion Basis Learning for Unsupervised Deep Homography Estimation with Subspace Projection".
BorderDet Image BorderDet: Border Feature for Dense Object Detection(ECCV2020 Oral)
CADDM Image Official implementation of ID-unaware Deepfake Detection Model
CR-DA-DET Image The official PyTorch implementation of paper Exploring Categorical Regularization for Domain Adaptive Object Detection (CR-DA-DET)
CREStereo Image Official MegEngine implementation of CREStereo(CVPR 2022 Oral).
CVPR2023-DMVFN Image CVPR2023 (highlight) - A Dynamic Multi-Scale Voxel Flow Network for Video Prediction
CVPR2023-UniDistill Image UniDistill: A Universal Cross-Modality Knowledge Distillation Framework for 3D Object Detection in Bird's-Eye View
CamLaserCalibraTool Image Extrinsic Calibration of a Camera and 2d Laser
CamOdomCalibraTool Image The tool to calibrate extrinsic param between camera and wheel
Co-mining Image Co-mining: Self-Supervised Learning for Sparsely Annotated Object Detection, AAAI 2021.
CoNR Image Official implementation of CoNR: Collaborative Neural Rendering using Anime Character Sheets
CrowdDetection Image None
D2C-SR Image Official MegEngine implementation of ECCV2022 "D2C-SR: A Divergence to Convergence Approach for Real-World Image Super-Resolution".
DCLS-SR Image Official PyTorch implementation of the paper "Deep Constrained Least Squares for Blind Image Super-Resolution", CVPR 2022.
DPGN Image This repository contains the official implementation of DPGN: Distribution Propagation Graph Network for Few-shot Learning.
DVN Image None
DeFCN Image End-to-End Object Detection with Fully Convolutional Network
DenseTeacher Image DenseTeacher: Dense Pseudo-Label for Semi-supervised Object Detection
DetNAS Image None
DisAlign Image Implementation of "Distribution Alignment: A Unified Framework for Long-tail Visual Recognition"(CVPR 2021)
DiverseBranchBlock Image None
Docs Image MegEngine Documentations
Documentation Image MegEngine Official Documentation
DynamicRouting Image Learning Dynamic Routing for Semantic Segmentation
ECCV2022-RIFE Image Official MegEngine Implementation of Real-Time Intermediate Flow Estimation for Video Frame Interpolation
ECCV2022-RIFE Image ECCV2022 - Real-Time Intermediate Flow Estimation for Video Frame Interpolation
ED-Net Image PyTorch implementation of A Lightweight Encoder-Decoder Path for Deep Residual Networks.
End-to-end-ASR-Transformer Image An end to end ASR Transformer model training repo
FINet Image This is the official MegEngine implementation of FINet: Dual Branches Feature Interaction for Partial-to-Partial Point Cloud Registration, AAAI 2022
FQ-ViT Image [IJCAI 2022] FQ-ViT: Post-Training Quantization for Fully Quantized Vision Transformer
FSCE Image None
FSSD_OoD_Detection Image Feature Space Singularity for Out-of-Distribution Detection. (SafeAI 2021)
FST-Matching Image Official implementation of the FST-Matching Model.
FullMatch Image Official implementation of FullMatch (CVPR2023)
FunnelAct Image None
GFSD Image This project provides an implementation for "Generalized Few-Shot Object Detection without Forgetting" (CVPR2021) on PyTorch.
GeneGAN Image Pytorch version of GeneGAN
GyroFlow Image The official MegEngine implementation of the ICCV 2021 paper: GyroFlow: Gyroscope-Guided Unsupervised Optical Flow Learning
HDR-Transformer Image The official MegEngine implementation of the ECCV 2022 paper: Ghost-free High Dynamic Range Imaging with Context-aware Transformer
HINet Image None
HomoGAN Image This is the official implementation of HomoGAN, CVPR2022
Hub Image 基于旷视研究院领先的深度学习算法，提供满足多业务场景的预训练模型
ICCV2019-LearningToPaint Image ICCV2019 - Learning to Paint With Model-based Deep Reinforcement Learning
ICD Image This is the official implementation of the paper "Instance-conditional Knowledge Distillation for Object Detection", based on MegEngine and Pytorch.
Inference-Demo Image 推理样例
Iter-E2EDET Image Official implementation of the paper "Progressive End-to-End Object Detection in Crowded Scenes"
KD-MVS Image Code for ECCV2022 paper 'KD-MVS: Knowledge Distillation Based Self-supervised Learning for Multi-view Stereo'
KPAFlow Image PyTorch implementation of KPA-Flow. Learning Optical Flow with Kernel Patch Attention (CVPR-2022)
LBHomo Image This is the official PyTorch implementation of Semi-supervised Deep Large-baseline Homography Estimation with Progressive Equivalence Constraint, AAAI 2023
LGD Image Official Implementation of the detection self-distillation framework LGD.
LLA Image Official implementation of our paper "LLA: Loss-aware Label Assignment for Dense Pedestrian Detection" in Pytorch.
LabelEnc Image None
MABN Image None
MEMD Image Megvii Electric Moped Detector (ONNX based inference)
ML-GCN Image PyTorch implementation of Multi-Label Image Recognition with Graph Convolutional Networks, CVPR 2019.
MM2022-ViCoPerceptualHeadGeneration Image MM2022 Workshop-Perceptual Conversational Head Generation with Regularized Driver and Enhanced Renderer
MOTR Image [ECCV2022] MOTR: End-to-End Multiple-Object Tracking with TRansformer
MOTRv2 Image [CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors
MSCL Image [ECCV2022] Motion Sensitive Contrastive Learning for Self-supervised Video Representation
MSPN Image Multi-Stage Pose Network
MegBA Image MegBA: A GPU-Based Distributed Library for Large-Scale Bundle Adjustment
MegCC Image MegCC是一个运行时超轻量，高效，移植简单的深度学习模型编译器
MegCat Image A Deep Learning Project about cats.
MegDiffusion Image MegEngine implementation of Diffusion Models.
MegEngine-Benchmark Image None
MegEngine-DMVFN Image None
MegEngine Image MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架
MegFlow Image Efficient ML solution for long-tailed demands.
MegPeak Image None
MegRay Image A communication library for deep learning
MegSpot Image MegSpot是一款高效、专业、跨平台的图片&视频对比应用
Megvii-BaseDetection Image You are welcomed to join us!
MetaPruning Image None
Models Image 采用MegEngine实现的各种主流深度学习模型
NAFNet Image The state-of-the-art image restoration model without nonlinear activation functions.
NBNet Image NBNet: Noise Basis Learning for Image Denoising with Subspace Projection
NIPS2017-LearningToRunACE Image 2nd place solution of NIPS2017 LearningToRun Competition.
NeRF Image NeRF implementation in MegEngine
NeurIPS2021-ML4CO-KIDA Image 1st Solution For NeurIPS 2021 Competition on ML4CO Dual Task
OMNet Image OMNet: Learning Overlapping Mask for Partial-to-Partial Point Cloud Registration, ICCV 2021, MegEngine implementation.
OTA Image Official implementation of our CVPR2021 paper "OTA: Optimal Transport Assignment for Object Detection" in Pytorch.
OccDepth Image Maybe the first academic open work on stereo 3D SSC method with vision-only input.
OdomLaserCalibraTool Image Extrinsic Calibration of a Odom and 2d Laser
PCB Image Official code for CVPR 2022 paper "Relieving Long-tailed Instance Segmentation via Pairwise Class Balance".
PETR Image [ECCV2022] PETR: Position Embedding Transformation for Multi-View 3D Object Detection
PMN Image [ACMMM 2022 Best Paper Runner-Up Award] Learnability Enhancement for Low-light Raw Denoising: Where Paired Real Data Meets Noise Modeling
PMRID Image ECCV2020 - Practical Deep Raw Image Denoising on Mobile Devices
Portraits_Correction Image None
RG-SENet_SP-SENet Image PyTorch implementation of Delving Deep into Spatial Pooling for Squeeze-and-Excitation Networks.
RLNAS Image None
RealFlow Image The official implementation of the ECCV 2022 Oral paper: RealFlow: EM-based Realistic Optical Flow Dataset Generation from Videos
RepLKNet Image Official MegEngine implementation of RepLKNet
RepVGG Image None
Resource Image None
RevCol Image Official Code of Paper "Reversible Column Networks"
SMP Image None
SOLQ Image "SOLQ: Segmenting Objects by Learning Queries", SOLQ is an end-to-end instance segmentation framework with Transformer.
SSQL-ECCV2022 Image PyTorch implementation of SSQL (Accepted to ECCV2022 oral presentation)
ShuffleNet-Series Image None
SimpleDG Image This is the training and test code for ECCV2022 workshop NICO challenge
SinglePathOneShot Image None
Sobolev_INRs Image [ECCV 2022] The official experimental code of "Sobolev Training for Implicit Neural Representations with Approximated Image Derivatives"
Sparsebit Image A model compression and acceleration toolbox based on pytorch.
TLC Image Test-time Local Converter
TP-LSD Image Official implementation of paper "TP-LSD: Tri-points based line segment detector" .
TPS-CVPR2023 Image None
TransMVSNet Image (CVPR 2022) TransMVSNet: Global Context-aware Multi-view Stereo Network with Transformers.
Transfer-Learning-Library Image Transfer Learning Library for Domain Adaptation, Task Adaptation, and Domain Generalization
TreeEnergyLoss Image [CVPR2022] Tree Energy Loss: Towards Sparsely Annotated Semantic Segmentation
TreeFilter-Torch Image Learnable Tree Filter for Structure-preserving Feature Transform
US3L-CVPR2023 Image PyTorch implementation of US3L (Accepted to CVPR2023)
WeightNet Image None
YOLOF Image None
YOLOX Image MegEngine implementation of YOLOX
YOLOX Image YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
awesome-megengine Image Awesome Resources about MegEngine
basecls Image A codebase & model zoo for pretrained backbone based on MegEngine.
basecls Image A codebase & model zoo for pretrained backbone based on MegEngine.
basecore Image basecore is a simple repo that provides deep learning frame for MegEngine.
basedet Image An object detection codebase based on MegEngine.
cheat_sheet_for_pytorch_immigrant Image 一份给从 PyTorch 迁移过来的用户的小抄
cpp_redis Image C++11 Lightweight Redis client: async, thread-safe, no dependency, pipelining, multi-platform
cpuinfo Image CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)
cudnn-frontend Image None
cutlass-bak Image modified cutlass
cutlass Image CUDA Templates for Linear Algebra Subroutines
cv-master-ex Image torch version of instant-ngp, image rendering
cvpods Image All-in-one Toolbox for Computer Vision Research.
examples Image A set of examples around MegEngine
flatbuffers Image FlatBuffers: Memory Efficient Serialization Library
gflags Image The gflags package contains a C++ library that implements commandline flags processing. It includes built-in support for standard types such as string and the ability to define flags in the source file in which they are used. Online documentation available at:
hpargparse Image argparse extension for hpman
hpman Image A hyperparameter manager for deep learning experiments.
hpnevergrad Image A nevergrad extension for hpman
introduction-neural-3d-reconstruction Image Course materials for Introduction to Neural 3D Reconstruction
invis Image invisible megengine API
juicefs-python Image JuiceFS Python SDK
mdistiller Image The official implementation of [CVPR2022] Decoupled Knowledge Distillation https://arxiv.org/abs/2203.08679
megengine-face-recognition Image None
megenginelite-rs Image None
megfile Image Megvii FILE Library - Working with Files in Python
megvii-pku-dl-course Image Homepage for the joint course of Megvii Inc. and Peking University on Deep Learning.
megvii-tsinghua-dl-course Image Slides with modifications for a course at Tsinghua University.
mgeconvert Image MegEngine到其他框架的转换器
midout Image Reduce binary size by removing code blocks
mperf-libpfm4 Image forked from https://sourceforge.net/p/perfmon2/libpfm4/ci/master/tree/
mperf Image mperf是一个面向移动/嵌入式平台的算子性能调优工具箱
neural-painter Image Paint artistic patterns using random neural network.
oneDNN Image oneAPI Deep Neural Network Library (oneDNN)
protoclip Image Official pytorch implementation of ProtoCLIP in paper Prototypical Contrastive Language Image Pretraining
pydata-sphinx-theme Image Bootstrap-based sphinx theme from the PyData community
pytorch-gym Image Implementation of the Deep Deterministic Policy Gradient(DDPG) in bullet Gym using pytorch
revisitAIRL Image [ECCV2022] Revisiting the Critical Factors of Augmentation-Invariant Representation Learning
storage Image provide Checkpoint for users.
swin-transformer Image Swin-Transformer implementation in MegEngine. This is a showcase for training on GPU with less memory by leveraging MegEngine DTR technique.
tf-cpn Image Cascade Pyramid Netwrok
tf-tutorials Image Tutorials for deep learning course here:
torch2mge Image None
video_analyst Image A series of basic algorithms that are useful for video understanding, including Single Object Tracking (SOT), Video Object Segmentation (VOS) and so on.
xopr Image Experimental Operator Library for MegEngine
zipfls Image This repo is the official megengine implementation of the ECCV2022 paper: Efficient One Pass Self-distillation with Zipf's Label Smoothing.

Pinned Loading

  1. NAFNet NAFNet Public

    The state-of-the-art image restoration model without nonlinear activation functions.

    Python 2.8k 364

  2. PETR PETR Public

    [ECCV2022] PETR: Position Embedding Transformation for Multi-View 3D Object Detection & [ICCV2023] PETRv2: A Unified Framework for 3D Perception from Multi-Camera Images

    Python 1k 152

  3. CREStereo CREStereo Public

    Official MegEngine implementation of CREStereo(CVPR 2022 Oral).

    Python 581 63

  4. Sparsebit Sparsebit Public

    A model compression and acceleration toolbox based on pytorch.

    Python 333 38

Repositories

Showing 10 of 92 repositories
View all repositories