WasedaMagina

SimMIM

This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".

mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

Mask2Former

Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"

vtpack

code base for vision transformers

CenterNet2

Two-stage CenterNet

ConditionalDETR

This repository is an official implementation of the ICCV 2021 paper "Conditional DETR for Fast Training Convergence". (https://arxiv.org/abs/2108.06152)

TOOD

TOOD: Task-aligned One-stage Object Detection, ICCV2021 Oral

DI-engine

OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.

YOLOX

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

T2T-ViT

ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet

Transformer-Explainability

[CVPR 2021] Official PyTorch implementation for Transformer Interpretability Beyond Attention Visualization, a novel method to visualize classifications by Transformer based networks.

pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

SegFormer

Official PyTorch implementation of SegFormer

Keras-MNIST-center-loss-with-visualization

An implementation for mnist center loss training and visualization

mnist_tsne

t-sne visualization of mnist images when feature is represented by raw pixels and cnn learned feature

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

vision_transformer

vision-transformer-pytorch

Pytorch version of Vision Transformer (ViT) with pretrained models. This is part of CASL (https://casl-project.github.io/) and ASYML project.

PVT

Official implementation of PVT series

Swin-Transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

TreeFilter-Torch

Learnable Tree Filter for Structure-preserving Feature Transform

Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.

SparseR-CNN

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

PaddleDetection

Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.

Relation-Networks-for-Object-Detection

Relation Networks for Object Detection

BorderDet

BorderDet: Border Feature for Dense Object Detection(ECCV2020 Oral)

detr

End-to-End Object Detection with Transformers

pytorch-cifar

95.47% on CIFAR10 with PyTorch

detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.