We write your reusable computer vision tools. 💜

Framework agnostic sliced/tiled inference + interactive ui + error analysis plots

Tensorflow Faster RCNN for Object Detection

Mask RCNN in TensorFlow

:pencil2: Web-based image segmentation tool for object detection, localization, and keypoints

[CVPR 2023] OneFormer: One Transformer to Rule Universal Image Segmentation

A coding-free framework built on PyTorch for reproducible deep learning studies. PyTorch Ecosystem. 🏆26 knowledge distillation methods presented at TPAMI, CVPR, ICLR, ECCV, NeurIPS, ICCV, AAAI, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

Real-time and accurate open-vocabulary end-to-end object detection

Two-stage CenterNet

Convert JSON annotations into YOLO format.

convert dataset to coco/voc format

PyTorch re-implementation of DeepLab v2 on COCO-Stuff / PASCAL VOC datasets

To speedup and simplify image labeling/ annotation process with multiple supported formats.

[ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention

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

Object Detection

Helper functions to create COCO datasets

Visual Question Answering in Pytorch

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)

Dataset Management Framework, a Python library and a CLI tool to build, analyze and manage Computer Vision datasets.

仿coco点餐系统的微信小程序

Keras beit,caformer,CMT,CoAtNet,convnext,davit,dino,efficientdet,edgenext,efficientformer,efficientnet,eva,fasternet,fastervit,fastvit,flexivit,gcvit,ghostnet,gpvit,hornet,hiera,iformer,inceptionnext,lcnet,levit,maxvit,mobilevit,moganet,nat,nfnets,pvt,swin,tinynet,tinyvit,uniformer,volo,vanillanet,yolor,yolov7,yolov8,yolox,gpt2,llama2, alias kecam

Computer vision based ML training data generation tool :rocket:

Object detection on multiple datasets with an automatically learned unified label space.

[TPAMI 2022 & CVPR2021 Oral] UP-DETR: Unsupervised Pre-training for Object Detection with Transformers

Implementation of various human pose estimation models in pytorch on multiple datasets (MPII & COCO) along with pretrained models

Official PyTorch implementation of Fully Attentional Networks

[ICML 2023] Official PyTorch implementation of Global Context Vision Transformers

various cv tools, such as label tools, data augmentation, label conversion, etc.