amsword

ViLT

Code for the ICML 2021 (long talk) paper: "ViLT: Vision-and-Language Transformer Without Convolution or Region Supervision"

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

maskrcnn-benchmark

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

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

setup

amsword.github.io

CMC

pytorch implementation of "Contrastive Multiview Coding", "Momentum Contrast for Unsupervised Visual Representation Learning", and "Unsupervised Feature Learning via Non-Parametric Instance-level Discrimination"

image_text_contrastive

image-text contrastive loss implementation

AutoAugment

Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow

azfuse

A lightweight blobfuse-like python tool with the data transfer through azcopy

BBox-Label-Tool

A simple tool for labeling object bounding boxes in images

caffe-1

Caffe: a fast open framework for deep learning.

clean-fid

PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]

cocoapi

COCO API - Dataset @ http://cocodataset.org/

DeepSpeed

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

detectron2

Detectron2 is FAIR's next-generation research platform for object detection and segmentation.

FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)

flash-attention

Fast and memory-efficient exact attention

fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.

gossipnet

Non-maximum suppression for object detection in a neural network

hnms

Hashing-based Non-Maximum Suppression

mmcv

Open MMLab Computer Vision Foundation

mmdetection

Open MMLab Detection Toolbox with PyTorch

pycocoevalcap

Python 3 support for the MS COCO caption evaluation tools

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-OpCounter

Count the FLOPs of your PyTorch model.

qdv

scaleapex

Fairscale + Apex

ssd.pytorch

A PyTorch Implementation of Single Shot MultiBox Detector

vim-snippets

vim-snipmate default snippets (Previously snipmate-snippets)