thebestyezhang Goto Github PK

2021

awesome-attention-mechanism-in-cv

:punch: CV中常用注意力模块;即插即用模块;ViT模型. PyTorch Implementation Collection of Attention Module and Plug&Play Module

captcha.pytorch

验证码识别比赛code，针对数字，大小写字母能达到97-99%的准确率

clip

Contrastive Language-Image Pretraining

cotnet

This is an official implementation for "Contextual Transformer Networks for Visual Recognition".

crossvit

Official implementation of CrossViT. https://arxiv.org/abs/2103.14899

cycle-gan-with-unet-as-generator

data_ans

db

A PyTorch implementation of "Real-time Scene Text Detection with Differentiable Binarization".

deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.

deit

Official DeiT repository

epsanet

lichee

一个多模态内容理解算法框架，其中包含数据处理、预训练模型、常见模型以及模型加速等模块。

loupe_keras

Rewrite the LOUPE library (https://github.com/antoine77340/LOUPE) into Keras version. Many learnable pooling or differentiable aggregation methods are covered. (NetVLAD, NetRVLAD, SoftDBoW, NetFV, CG)

pytorch-gan

PyTorch implementations of Generative Adversarial Networks.

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

pytorch-unet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

swin-transformer

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

transgan

This is a re-implementation of TransGAN: Two Pure Transformers Can Make One Strong GAN (CVPR 2021) in PyTorch.

video-swin-transformer

This is an official implementation for "Video Swin Transformers".

vit-pytorch

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