🍅🍅🍅YOLOv5-Lite: lighter, faster and easier to deploy. Evolved from yolov5 and the size of model is only 930+kb (int8) and 1.7M (fp16). It can reach 10+ FPS on the Raspberry Pi 4B when the input size is 320×320~

✭ 1,230

python C++Makefile c CMake shell pytorch transformer android-app tensorrt mnn mobilenet ncnn tflite shufflenetv2 onnxruntime yolov5 repvgg pplcnet openvivo picodet

TransPose

PyTorch Implementation for "TransPose: Keypoint localization via Transformer", ICCV 2021.

✭ 250

Jupyter Notebook Cuda python localization transformer pose-estimation keypoint

transformer

A PyTorch Implementation of "Attention Is All You Need"

✭ 28

python nlp transformer seq2seq attention attention-is-all-you-need

BossNAS

(ICCV 2021) BossNAS: Exploring Hybrid CNN-transformers with Block-wisely Self-supervised Neural Architecture Search

✭ 125

python shell transformer nas automl neural-architecture-search self-supervised-learning hybrid-neural-network

COVID-19-Tweet-Classification-using-Roberta-and-Bert-Simple-Transformers

Rank 1 / 216

✭ 24

Jupyter Notebook sentiment-analysis text-classification transformers transformer classification tweet bert sentiment-classification textclassification roberta covid-19 covid19

OpenPrompt

An Open-Source Framework for Prompt-Learning.

✭ 1,769

python shell nlp natural-language-processing ai deep-learning prompt pytorch transformer prompt-toolkit nlp-library nlp-machine-learning prompts natural-language-understanding pre-trained-model pre-trained-language-models prompt-based-tuning prompt-learning

graph-transformer-pytorch

Implementation of Graph Transformer in Pytorch, for potential use in replicating Alphafold2

✭ 81

python deep-learning graphs artificial-intelligence transformer

GTSRB Keras STN

German Traffic Sign Recognition Benchmark, Keras implementation with Spatial Transformer Networks

✭ 48

Jupyter Notebook python network keras spatial transformer gtsrb stn

cometa

Corpus of Online Medical EnTities: the cometA corpus

✭ 31

Jupyter Notebook python nlp machine-learning deep-learning transformer entity-linking bionlp

pytorch-gpt-x

Implementation of autoregressive language model using improved Transformer and DeepSpeed pipeline parallelism.

✭ 21

python shell pytorch transformer gpt pipeline-parallelism deepspeed

NiuTrans.NMT

A Fast Neural Machine Translation System. It is developed in C++ and resorts to NiuTensor for fast tensor APIs.

✭ 112

C++python CMake machine-translation transformer neural-machine-translation fast-decoding

YOLOS

You Only Look at One Sequence (NeurIPS 2021)

✭ 612

Jupyter Notebook python computer-vision transformer object-detection vision-transformer

verseagility

Ramp up your custom natural language processing (NLP) task, allowing you to bring your own data, use your preferred frameworks and bring models into production.

✭ 23

Jupyter Notebook python typescript nlp transformer classification question-answering summarization ner machine-reading-comprehension dstoolkit

transformer-tensorflow2.0

transformer in tensorflow 2.0

✭ 53

Jupyter Notebook python perl shell tensorflow tf2 transformer tensorflow2 transformer-tensorflow2

zero

Zero -- A neural machine translation system

✭ 121

python perl transformer neural-machine-translation average-attention-network aan speech-translation depth-scaled-initialization deep-transformer l0drop adaptive-feature-selection massively-multilingual-translation opus-100 fast-bidirectional-decoder

TransMorph Transformer for Medical Image Registration

TransMorph: Transformer for Unsupervised Medical Image Registration (PyTorch)

✭ 130

python Jupyter Notebook deep-learning transformer diffeomorphism image-registration bayesian-deep-learning image-alignment vision-transformer swin-transformer

wenet

Production First and Production Ready End-to-End Speech Recognition Toolkit

✭ 2,384

C++python shell perl CMake java pytorch transformer speech-recognition automatic-speech-recognition production-ready asr conformer e2e-models

Quality-Estimation1

机器翻译子任务-翻译质量评价-复现 WMT2018 阿里论文结果

✭ 19

python perl shell Jupyter Notebook transformer nmt quality-estimation wmt

MusicTransformer-Pytorch

MusicTransformer written for MaestroV2 using the Pytorch framework for music generation

✭ 106

python mit pytorch transformer maestro music-generation music-transformer

dodrio

Exploring attention weights in transformer-based models with linguistic knowledge.

✭ 233

Svelte javascript python visualization nlp deep-learning transformer attention-mechanism interactive-visualizations multi-head-attention

wxml-transformer

将微信小程序的wxml代码转换成js object或html片段

✭ 18

javascript transformer wxml wxapp wxml-transformer wxml-to-html

Conformer

Official code for Conformer: Local Features Coupling Global Representations for Visual Recognition

✭ 345

Jupyter Notebook python cnn transformer classification object-detection

AdaSpeech

AdaSpeech: Adaptive Text to Speech for Custom Voice

✭ 108

Jupyter Notebook python text-to-speech speech pytorch tts speech-synthesis transformer pytorch-implementation fastspeech fastspeech2 adaspeech

Context-Transformer

Context-Transformer: Tackling Object Confusion for Few-Shot Detection, AAAI 2020

✭ 89

python c Cuda pytorch transformer object-detection transfer-learning few-shot rfbnet

Transformer-Transducer

PyTorch implementation of "Transformer Transducer: A Streamable Speech Recognition Model with Transformer Encoders and RNN-T Loss" (ICASSP 2020)

✭ 61

python end-to-end transformer speech-recognition sequence-to-sequence rnnt transformer-transducer

set-transformer

A neural network architecture for prediction on sets

✭ 18

python Jupyter Notebook pytorch transformer

paccmann proteomics

PaccMann models for protein language modeling

✭ 28

python shell pytorch transformer protein-language-model

Kevinpro-NLP-demo

All NLP you Need Here. 个人实现了一些好玩的NLP demo，目前包含13个NLP应用的pytorch实现

✭ 117

python Jupyter Notebook nlp natural-language-processing text-classification pytorch transformer baseline bert vae-gan textclassification

ICON

(TPAMI2022) Salient Object Detection via Integrity Learning.

✭ 125

python shell deep-learning transformer codebase sod saliency mlp-mixer

NLP-paper

🎨 🎨NLP 自然语言处理教程 🎨🎨 https://dataxujing.github.io/NLP-paper/

✭ 23

word2vec crf pagerank textrank transformer seq2seq lsa glove gpt lda fasttext attention-mechanism albert bert textcnn elmo seq2seq-attention plsa xlnet lad2vec

SegFormer

Official PyTorch implementation of SegFormer

✭ 1,264

python transformer semantic-segmentation cityscapes ade20k

CrabNet

Predict materials properties using only the composition information!

✭ 57

python machine-learning scikit-learn pytorch transformer neural-networks attention materials-science materials-informatics attention-mechanism materials-screening self-attention predict-materials-properties

semantic-segmentation

SOTA Semantic Segmentation Models in PyTorch

✭ 464

python Jupyter Notebook deep-learning pytorch dataset transformer semantic-segmentation cityscapes ade20k pascal-context camvid vision-transformer coco-stuff

TDRG

Transformer-based Dual Relation Graph for Multi-label Image Recognition. ICCV 2021

✭ 32

python transformer gcn multi-label-image-classification

sticker2

Further developed as SyntaxDot: https://github.com/tensordot/syntaxdot

✭ 14

rust Nix python pytorch transformer bert part-of-speech-tagger dependency-parsing topological-fields

learningspoons

nlp lecture-notes and source code

✭ 29

Jupyter Notebook python nlp tensorflow word2vec cnn transformer lstm ipynb attention cbow lecture-notes seq2seq-model attention-model colab-notebook

Graphormer

Graphormer is a deep learning package that allows researchers and developers to train custom models for molecule modeling tasks. It aims to accelerate the research and application in AI for molecule science, such as material design, drug discovery, etc.

✭ 1,194

python cython shell deep-learning graph transformer ai4science molecule-simulation

KitanaQA

KitanaQA: Adversarial training and data augmentation for neural question-answering models

✭ 58

python nlp natural-language-processing pytorch transformer question-answering bert data-augmentation adversarial-attacks adversarial-training ml-automation

h-transformer-1d

Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning

✭ 121

python deep-learning artificial-intelligence transformer attention attention-mechanism

tensorflow-ml-nlp-tf2

텐서플로2와 머신러닝으로 시작하는 자연어처리 (로지스틱회귀부터 BERT와 GPT3까지) 실습자료