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

✏️ Web-based image segmentation tool for object detection, localization, and keypoints

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

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

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

Two-stage CenterNet

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

convert dataset to coco/voc format

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

Convert JSON annotations into YOLO format.

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

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

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)

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