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

  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 875 132

  3. CREStereo CREStereo Public

    Official MegEngine implementation of CREStereo(CVPR 2022 Oral).

    Python 497 61

  4. Sparsebit Sparsebit Public

    A model compression and acceleration toolbox based on pytorch.

    Python 327 40

Repositories

Showing 10 of 90 repositories
  • megfile Public

    Megvii FILE Library - Working with Files in Python same as the standard library

    megvii-research/megfile’s past year of commit activity
    Python 128 Apache-2.0 17 3 1 Updated Nov 22, 2024
  • MCTrack Public

    This is the offical implementation of the paper "MCTrack: A Unified 3D Multi-Object Tracking Framework for Autonomous Driving"

    megvii-research/MCTrack’s past year of commit activity
    Python 81 4 2 0 Updated Nov 16, 2024
  • megactor Public
    megvii-research/megactor’s past year of commit activity
    Python 819 Apache-2.0 118 20 0 Updated Oct 21, 2024
  • Sparse-Beats-Dense Public

    [ECCV 2024] Sparse Beats Dense: Rethinking Supervision in Radar-Camera Depth Completion

    megvii-research/Sparse-Beats-Dense’s past year of commit activity
    Python 21 Apache-2.0 0 2 0 Updated Oct 9, 2024
  • CasPL Public
    megvii-research/CasPL’s past year of commit activity
    28 Apache-2.0 1 3 0 Updated Oct 8, 2024
  • HiDiffusion Public

    [ECCV 2024] HiDiffusion: Increases the resolution and speed of your diffusion model by only adding a single line of code!

    megvii-research/HiDiffusion’s past year of commit activity
    Jupyter Notebook 773 Apache-2.0 44 18 1 Updated Aug 15, 2024
  • DPGN Public

    [CVPR 2020] DPGN: Distribution Propagation Graph Network for Few-shot Learning.

    megvii-research/DPGN’s past year of commit activity
    Python 181 MIT 38 18 2 Updated Jul 25, 2024
  • NAFNet Public

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

    megvii-research/NAFNet’s past year of commit activity
    Python 2,263 283 100 4 Updated Jul 3, 2024
  • LBHomo Public

    This is the official PyTorch implementation of Semi-supervised Deep Large-baseline Homography Estimation with Progressive Equivalence Constraint, AAAI 2023

    megvii-research/LBHomo’s past year of commit activity
    Python 18 3 8 0 Updated Jun 18, 2024
  • .github Public
    megvii-research/.github’s past year of commit activity
    2 1 0 0 Updated Apr 19, 2024
View all repositories