songlilucky

qlib

Qlib is an AI-oriented quantitative investment platform that aims to realize the potential, empower research, and create value using AI technologies in quantitative investment, from exploring ideas to implementing productions. Qlib supports diverse machine learning modeling paradigms. including supervised learning, market dynamics modeling, and RL.

pytorch3d

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

nanodet

NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥

OpenPCDet

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.

kaolin

A PyTorch Library for Accelerating 3D Deep Learning Research

deit

Official DeiT repository

mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

ResNeSt

ResNeSt: Split-Attention Networks

Objectron

Objectron is a dataset of short, object-centric video clips. In addition, the videos also contain AR session metadata including camera poses, sparse point-clouds and planes. In each video, the camera moves around and above the object and captures it from different views. Each object is annotated with a 3D bounding box. The 3D bounding box describes the object’s position, orientation, and dimensions. The dataset contains about 15K annotated video clips and 4M annotated images in the following categories: bikes, books, bottles, cameras, cereal boxes, chairs, cups, laptops, and shoes

PyTorch-Encoding

A CV toolkit for my papers.

ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network

once-for-all

[ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment

CenterPoint

SparseR-CNN

[CVPR2021, PAMI2023] End-to-End Object Detection with Learnable Proposal

CSBook

PolarMask

Code for 'PolarMask: Single Shot Instance Segmentation with Polar Representation'

SegmenTron

Support PointRend, Fast_SCNN, HRNet, Deeplabv3_plus(xception, resnet, mobilenet), ContextNet, FPENet, DABNet, EdaNet, ENet, Espnetv2, RefineNet, UNet, DANet, HRNet, DFANet, HardNet, LedNet, OCNet, EncNet, DuNet, CGNet, CCNet, BiSeNet, PSPNet, ICNet, FCN, deeplab)

PCT

Jittor implementation of PCT:Point Cloud Transformer

WoodScape

The repository containing tools and information about the WoodScape dataset.

Pointnet2.PyTorch

A faster implementation of PointNet++ based on PyTorch.

epipolar-transformers

Epipolar Transformers (best paper award, CVPR 2020 workshop)

awesome-anchor-free-object-detection

In this project, we focus on collection the anchor free object detection paper or code.

qdtrack

Quasi-Dense Similarity Learning for Multiple Object Tracking, CVPR 2021 (Oral)

swa_object_detection

SWA Object Detection

Video-Action-Transformer-Network-Pytorch-

Implementation of the paper Video Action Transformer Network

ZoomNet

ZoomNet: Part-Aware Adaptive Zooming Neural Network for 3D Object Detection

lambda.pytorch

PyTorch implementation of Lambda Network and pretrained Lambda-ResNet

WaymoCOCO

Waymo Open Dataset converter to COCO format

Stereo-3D-Object-Detection-for-Autonomous-Driving

3D object detection using Stereo-RCNN for Autonomous Driving

3D-NMS-and-3D-IoU