MEGVII Research (@megvii-research)
Guthub

megvii-research

Twitter logo Share LinkedIn logo Share Facebook logo Share

Top repositories

1

NAFNet

The state-of-the-art image restoration model without nonlinear activation functions.
Python
1,998
star
2

ML-GCN

PyTorch implementation of Multi-Label Image Recognition with Graph Convolutional Networks, CVPR 2019.
Python
1,384
star
3

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.
Python
811
star
4

IJCAI2023-CoNR

IJCAI2023 - Collaborative Neural Rendering using Anime Character Sheets
Jupyter Notebook
782
star
5

PETR

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

mdistiller

The official implementation of [CVPR2022] Decoupled Knowledge Distillation https://arxiv.org/abs/2203.08679 and [ICCV2023] DOT: A Distillation-Oriented Trainer https://openaccess.thecvf.com/content/ICCV2023/papers/Zhao_DOT_A_Distillation-Oriented_Trainer_ICCV_2023_paper.pdf
Python
739
star
7

BBN

The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition
Python
656
star
8

MOTR

[ECCV2022] MOTR: End-to-End Multiple-Object Tracking with TRansformer
Python
553
star
9

neural-painter

Paint artistic patterns using random neural network.
Python
531
star
10

CREStereo

Official MegEngine implementation of CREStereo(CVPR 2022 Oral).
Python
443
star
11

megvii-pku-dl-course

Homepage for the joint course of Megvii Inc. and Peking University on Deep Learning.
Python
439
star
12

HiDiffusion

Jupyter Notebook
384
star
13

MSPN

Multi-Stage Pose Network
Python
333
star
14

MOTRv2

[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors
Python
326
star
15

AnchorDETR

An official implementation of the Anchor DETR.
Python
321
star
16

Sparsebit

A model compression and acceleration toolbox based on pytorch.
Python
319
star
17

FQ-ViT

[IJCAI 2022] FQ-ViT: Post-Training Quantization for Fully Quantized Vision Transformer
Python
273
star
18

FSCE

Python
272
star
19

OccDepth

Maybe the first academic open work on stereo 3D SSC method with vision-only input.
Python
258
star
20

TransMVSNet

(CVPR 2022) TransMVSNet: Global Context-aware Multi-view Stereo Network with Transformers.
Python
258
star
21

RevCol

Official Code of Paper "Reversible Column Networks" "RevColv2"
Python
243
star
22

DCLS-SR

Official PyTorch implementation of the paper "Deep Constrained Least Squares for Blind Image Super-Resolution", CVPR 2022.
Python
215
star
23

TLC

Test-time Local Converter
Python
214
star
24

SOLQ

"SOLQ: Segmenting Objects by Learning Queries", SOLQ is an end-to-end instance segmentation framework with Transformer.
Python
195
star
25

introduction-neural-3d-reconstruction

Course materials for Introduction to Neural 3D Reconstruction
182
star
26

AAAI2023-PVD

Official Implementation of PVD and PVDAL: http://sk-fun.fun/PVD-AL/
Python
180
star
27

tf-tutorials

Tutorials for deep learning course here:
Jupyter Notebook
180
star
28

DPGN

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

PMN

[TPAMI 2023 / ACMMM 2022 Best Paper Runner-Up Award] Learnability Enhancement for Low-light Raw Denoising: Where Paired Real Data Meets Noise Modeling (a Data Perspective)
Python
120
star
30

CADDM

Official implementation of ID-unaware Deepfake Detection Model
C++
116
star
31

CR-DA-DET

The official PyTorch implementation of paper Exploring Categorical Regularization for Domain Adaptive Object Detection (CR-DA-DET)
Python
115
star
32

megfile

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

TreeEnergyLoss

[CVPR2022] Tree Energy Loss: Towards Sparsely Annotated Semantic Segmentation
Python
101
star
34

Far3D

[AAAI2024] Far3D: Expanding the Horizon for Surround-view 3D Object Detection
Jupyter Notebook
98
star
35

CVPR2023-UniDistill

CVPR2023 (highlight) - UniDistill: A Universal Cross-Modality Knowledge Distillation Framework for 3D Object Detection in Bird's-Eye View
Python
96
star
36

hpman

A hyperparameter manager for deep learning experiments.
Python
94
star
37

RealFlow

The official implementation of the ECCV 2022 Oral paper: RealFlow: EM-based Realistic Optical Flow Dataset Generation from Videos
Python
85
star
38

Iter-E2EDET

Official implementation of the paper "Progressive End-to-End Object Detection in Crowded Scenes"
Python
84
star
39

HDR-Transformer

The official MegEngine implementation of the ECCV 2022 paper: Ghost-free High Dynamic Range Imaging with Context-aware Transformer
Python
82
star
40

FSSD_OoD_Detection

Feature Space Singularity for Out-of-Distribution Detection. (SafeAI 2021)
Python
80
star
41

cv-master-ex

torch version of instant-ngp, image rendering
C++
78
star
42

SSQL-ECCV2022

PyTorch implementation of SSQL (Accepted to ECCV2022 oral presentation)
Python
75
star
43

expman

Shell
62
star
44

megvii-tsinghua-dl-course

Slides with modifications for a course at Tsinghua University.
56
star
45

BasesHomo

The official PyTorch implementation of the paper "Motion Basis Learning for Unsupervised Deep Homography Estimation with Subspace Projection".
Python
53
star
46

LGD

Official Implementation of the detection self-distillation framework LGD.
Python
52
star
47

D2C-SR

Official MegEngine implementation of ECCV2022 "D2C-SR: A Divergence to Convergence Approach for Real-World Image Super-Resolution".
Python
43
star
48

KD-MVS

Code for ECCV2022 paper 'KD-MVS: Knowledge Distillation Based Self-supervised Learning for Multi-view Stereo'
Python
42
star
49

protoclip

📍 Official pytorch implementation of paper "ProtoCLIP: Prototypical Contrastive Language Image Pretraining" (IEEE TNNLS)
Python
41
star
50

pytorch-gym

Implementation of the Deep Deterministic Policy Gradient(DDPG) in bullet Gym using pytorch
Python
40
star
51

AGFlow

Learning Optical Flow with Adaptive Graph Reasoning (AGFlow, AAAI-2022)
Python
39
star
52

KPAFlow

PyTorch implementation of KPA-Flow. Learning Optical Flow with Kernel Patch Attention (CVPR-2022)
Python
37
star
53

FullMatch

Official implementation of FullMatch (CVPR2023)
Python
37
star
54

TPS-CVPR2023

Python
37
star
55

HomoGAN

This is the official implementation of HomoGAN, CVPR2022
Python
35
star
56

PCB

Official code for CVPR 2022 paper "Relieving Long-tailed Instance Segmentation via Pairwise Class Balance".
Python
34
star
57

FST-Matching

Official implementation of the FST-Matching Model.
Python
33
star
58

US3L-CVPR2023

PyTorch implementation of US3L (Accepted to CVPR2023)
Python
32
star
59

basecls

A codebase & model zoo for pretrained backbone based on MegEngine.
Python
32
star
60

Sobolev_INRs

[ECCV 2022] The official experimental code of "Sobolev Training for Implicit Neural Representations with Approximated Image Derivatives"
Python
29
star
61

Portraits_Correction

Python
29
star
62

basedet

An object detection codebase based on MegEngine.
Python
27
star
63

Co-mining

Co-mining: Self-Supervised Learning for Sparsely Annotated Object Detection, AAAI 2021.
Python
26
star
64

zipfls

This repo is the official megengine implementation of the ECCV2022 paper: Efficient One Pass Self-distillation with Zipf's Label Smoothing.
Python
24
star
65

tf-cpn

Cascade Pyramid Netwrok
Python
24
star
66

Arch-Net

Arch-Net: Model Distillation for Architecture Agnostic Model Deployment
Python
22
star
67

IntLLaMA

IntLLaMA: A fast and light quantization solution for LLaMA
Python
21
star
68

juicefs-python

JuiceFS Python SDK
Python
18
star
69

ED-Net

PyTorch implementation of A Lightweight Encoder-Decoder Path for Deep Residual Networks.
Python
18
star
70

MSCL

[ECCV2022] Motion Sensitive Contrastive Learning for Self-supervised Video Representation
Python
16
star
71

hpargparse

argparse extension for hpman
Python
16
star
72

RG-SENet_SP-SENet

PyTorch implementation of Delving Deep into Spatial Pooling for Squeeze-and-Excitation Networks.
Python
16
star
73

LBHomo

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

MEMD

Megvii Electric Moped Detector (ONNX based inference)
Python
13
star
75

DVN

Python
12
star
76

Occ2net

Jupyter Notebook
12
star
77

revisitAIRL

[ECCV2022] Revisiting the Critical Factors of Augmentation-Invariant Representation Learning
Python
11
star
78

megengine-face-recognition

Python
9
star
79

SimpleDG

This is the training and test code for ECCV2022 workshop NICO challenge
Python
7
star
80

GeneGAN

Pytorch version of GeneGAN
Python
7
star
81

basecore

basecore is a simple repo that provides deep learning frame for MegEngine.
Python
7
star
82

hpnevergrad

A nevergrad extension for hpman
Python
4
star
83

.github

2
star
84

DRConv

Python
1
star