sliu's starred repositories

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

Language:PythonLicense:Apache-2.0Stargazers:30254Issues:310Issues:874

mediapipe

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

Language:C++License:Apache-2.0Stargazers:25627Issues:491Issues:4907

nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

Language:PythonLicense:MITStargazers:13819Issues:284Issues:2069

detr

End-to-End Object Detection with Transformers

Language:PythonLicense:Apache-2.0Stargazers:12989Issues:150Issues:526

pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques

Language:PythonLicense:MITStargazers:7736Issues:115Issues:106

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

Language:PythonLicense:Apache-2.0Stargazers:7558Issues:54Issues:2296

automl

Google Brain AutoML

Language:Jupyter NotebookLicense:Apache-2.0Stargazers:6163Issues:152Issues:884

ORB_SLAM3

ORB-SLAM3: An Accurate Open-Source Library for Visual, Visual-Inertial and Multi-Map SLAM

Language:C++License:GPL-3.0Stargazers:6107Issues:126Issues:785

awesome-self-supervised-learning

A curated list of awesome self-supervised methods

Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.

Language:Jupyter NotebookLicense:LGPL-3.0Stargazers:5193Issues:108Issues:715

mmdetection3d

OpenMMLab's next-generation platform for general 3D object detection.

Language:PythonLicense:Apache-2.0Stargazers:4927Issues:62Issues:1563

moco

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

Language:PythonLicense:MITStargazers:4598Issues:52Issues:131

OpenPCDet

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.

Language:PythonLicense:Apache-2.0Stargazers:4393Issues:70Issues:1396

distiller

Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller

Language:Jupyter NotebookLicense:Apache-2.0Stargazers:4314Issues:132Issues:350

AdelaiDet

AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.

Language:PythonLicense:NOASSERTIONStargazers:3333Issues:84Issues:543

mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

Language:PythonLicense:Apache-2.0Stargazers:3229Issues:31Issues:752

pycls

Codebase for Image Classification Research, written in PyTorch.

Language:PythonLicense:MITStargazers:2115Issues:62Issues:81

RepDistiller

[ICLR 2020] Contrastive Representation Distillation (CRD), and benchmark of recent knowledge distillation methods

Language:PythonLicense:BSD-2-ClauseStargazers:2097Issues:17Issues:56

fast-autoaugment

Official Implementation of 'Fast AutoAugment' in PyTorch.

Language:PythonLicense:MITStargazers:1584Issues:42Issues:59

shape_based_matching

try to implement halcon shape based matching, refer to machine vision algorithms and applications, page 317 3.11.5, written by halcon engineers

Language:C++License:BSD-2-ClauseStargazers:1171Issues:59Issues:206

structure_knowledge_distillation

The official code for the paper 'Structured Knowledge Distillation for Semantic Segmentation'. (CVPR 2019 ORAL) and extension to other tasks.

Language:PythonLicense:BSD-2-ClauseStargazers:690Issues:14Issues:71

TSD

1st place models in Google OpenImage Detection Challenge 2019

Language:PythonLicense:Apache-2.0Stargazers:454Issues:29Issues:38

BorderDet

BorderDet: Border Feature for Dense Object Detection(ECCV2020 Oral)

Language:PythonLicense:Apache-2.0Stargazers:430Issues:19Issues:58

SEPC

Scale-equalizing Pyramid Convolution for object detection(CVPR2020)

Language:PythonLicense:Apache-2.0Stargazers:330Issues:14Issues:27

NAS-FCOS

NAS-FCOS: Fast Neural Architecture Search for Object Detection (CVPR 2020)

Language:PythonLicense:BSD-2-ClauseStargazers:187Issues:10Issues:13

DynamicRCNN

Dynamic R-CNN: Towards High Quality Object Detection via Dynamic Training, ECCV 2020

Language:PythonLicense:MITStargazers:173Issues:4Issues:10

apq

[CVPR 2020] APQ: Joint Search for Network Architecture, Pruning and Quantization Policy

Language:PythonLicense:NOASSERTIONStargazers:156Issues:10Issues:7

nas-segm-pytorch

Code for Fast Neural Architecture Search of Compact Semantic Segmentation Models via Auxiliary Cells, CVPR '19

Language:SmalltalkLicense:NOASSERTIONStargazers:151Issues:8Issues:6

AutoTorch

AutoTorch, A HPO Toolkit

Language:PythonLicense:Apache-2.0Stargazers:59Issues:8Issues:0