zeliu98 Goto Github PK

closerlook3d

A Closer Look at Local Aggregation Operators in Point Cloud Analysis(ECCV 2020)

deepspeed

DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.

deit

Official DeiT repository

detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

gcnet

GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond

group-free-3d

Group-Free 3D Object Detection via Transformers

incubator-mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more

kpconv

Kernel Point Convolutions

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

ms-amp

Microsoft Automatic Mixed Precision Library

pic

Parametric Instance Classification for Unsupervised Visual Feature Learning, NeurIPS 2020

pixpro

Propagate Yourself: Exploring Pixel-Level Consistency for Unsupervised Visual Representation Learning, CVPR 2021

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- (SE)ResNet/ResNeXT, DPN, EfficientNet, MixNet, MobileNet-V3/V2, MNASNet, Single-Path NAS, FBNet, and more

pytorch-memory-utils

pytorch memory track code

relation-shape-cnn

Relation-Shape Convolutional Neural Network for Point Cloud Analysis (CVPR 2019 Oral & Best paper finalist)

swin-transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

torch-points3d

Pytorch framework for doing deep learning on point clouds.

tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation

tutorials

PyTorch tutorials.

video-swin-transformer

This is an official implementation for "Video Swin Transformers".

vision

Datasets, Transforms and Models specific to Computer Vision

vision_transformer