wzb's repositories
AutoAugment
Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow
ScaledYOLOv4
Scaled-YOLOv4: Scaling Cross Stage Partial Network
gen-efficientnet-pytorch
Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS
Yolov5_DeepSort_Pytorch
Real-time multi-object tracker using YOLO v5 and deep sort
HotSpot
HotSpot is an accurate and fast thermal model suitable for use in architectural studies.
DynamicViT
DynamicViT: Efficient Vision Transformers with Dynamic Token Sparsification
transformers
🤗Transformers: State-of-the-art Natural Language Processing for Pytorch, TensorFlow, and JAX.
text-to-text-transfer-transformer
Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"
bert4keras
keras implement of transformers for humans
detr
End-to-End Object Detection with Transformers
noah-research
Noah Research
Swin-Transformer
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
SViTE
[Preprint] "Chasing Sparsity in Vision Transformers: An End-to-End Exploration" by Tianlong Chen, Yu Cheng, Zhe Gan, Lu Yuan, Lei Zhang, Zhangyang Wang
Awesome-Visual-Transformer
Collect some papers about transformer with vision. Awesome Transformer with Computer Vision (CV)
YOLOS
You Only Look at One Sequence (https://arxiv.org/abs/2106.00666)
fast-transformers
Pytorch library for fast transformer implementations
deit
Official DeiT repository
big_transfer
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
filter-pruning-geometric-median
Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Acceleration (CVPR 2019 Oral)
keras-retinanet
Keras implementation of RetinaNet object detection.
PyTorch-Encoding
A CV toolkit for my papers.
yolov3
YOLOv3 in PyTorch > ONNX > CoreML > TFLite
Semantic-Quantization
Tensorflow implementation for Training and Inference for Integer-based Semantic Segmentation
tensorflow-yolov3
🔥 TensorFlow Code for technical report: "YOLOv3: An Incremental Improvement"
TurboTransformers
a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.
google-landmark
Dataset with 5 million images depicting human-made and natural landmarks spanning 200 thousand classes.
gpt2-ml
GPT2 for Multiple Languages, including pretrained models. GPT2 多语言支持, 15亿参数中文预训练模型