coco

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

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

Legacy JSON-to-YOLO dataset converter for COCO, LabelMe, Labelbox, VoTT, INFOLKS, and ATH annotations. Superseded by convert_coco() in the Ultralytics package.

Two-stage CenterNet

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)