pretrained-models

There are 53 repositories under pretrained-models topic.

transformers
huggingface / transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
nlp natural-language-processing pytorch language-model tensorflow bert language-models pytorch-transformers nlp-library transformer model-hub pretrained-models jax flax seq2seq speech-recognition hacktoberfest python machine-learning deep-learning
Language:Python 86756
huggingface / 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 imagenet-classifier resnet dual-path-networks cnn-classification pretrained-models pretrained-weights distributed-training mobile-deep-learning mobilenet-v2 mnasnet mobilenetv3 efficientnet augmix randaugment mixnet efficientnet-training vision-transformer-models nfnets normalization-free-training
Language:Python 24243
spleeter
deezer / spleeter
Deezer source separation library including pretrained models.
audio-processing bass deep-learning deezer drums model pretrained-models python tensorflow vocals
Language:Python 21943
PaddleNLP
PaddlePaddle / PaddleNLP
👑 Easy-to-use and powerful NLP library with 🤗 Awesome model zoo, supporting wide-range of NLP tasks from research to industrial applications, including 🗂Text Classification, 🔍 Neural Search, ❓ Question Answering, ℹ️ Information Extraction, 📄 Document Intelligence, 💌 Sentiment Analysis and 🖼 Diffusion AIGC system etc.
aigc bert compression document-intelligence embedding ernie image-generation information-extraction neural-search nlp paddlenlp pretrained-models question-answering search-engine semantic-analysis sentiment-analysis seq2seq stable-diffusion transformers uie
Language:Python 8124
lukemelas / EfficientNet-PyTorch
A PyTorch implementation of EfficientNet and EfficientNetV2 (coming soon!)
efficientnet-pytorch feature-extraction imagenet pretrained-models
Language:Python 7331
qubvel / segmentation_models.pytorch
Segmentation models with pretrained backbones. PyTorch.
segmentation image-processing pspnet linknet unet unet-pytorch pytorch fpn models imagenet pretrained-models semantic-segmentation image-segmentation segmentation-models pretrained-backbones unetplusplus deeplabv3 deeplab-v3-plus pretrained-weights hacktoberfest
Language:Python 6963
microsoft / nlp-recipes
Natural Language Processing Best Practices & Examples
nlp natural-language-processing natural-language-understanding text deep-learning azure-ml nlu nli natural-language-inference mlflow machine-learning natural-language pretrained-models sota transfomer text-classification best-practices
Language:Python 6103
facebookresearch / mmf
A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)
pytorch vqa pretrained-models multimodal deep-learning captioning dialog textvqa hateful-memes multi-tasking
Language:Python 5167
PaddlePaddle / PaddleClas
A treasure chest for visual classification and recognition powered by PaddlePaddle
autoaugment cutmix deit fastdeploy gridmask image-classification image-recognition image-retrieval-system knowledge-distillation pretrained-models product-recognition randaugment repvgg swin-transformer
Language:Python 4717
mlfoundations / open_clip
An open source implementation of CLIP.
deep-learning pytorch computer-vision language-model multi-modal-learning contrastive-loss zero-shot-classification pretrained-models
Language:Python 3787
snakers4 / silero-models
Silero Models: pre-trained speech-to-text, text-to-speech and text-enhancement models made embarrassingly simple
speech-recognition speech-to-text stt asr pretrained-models english german spanish stt-benchmark pytorch colab onnx torch-hub text-to-speech tts-models speech speech-synthesis tts repunctuation capitalization
Language:Jupyter Notebook 3369
timesler / facenet-pytorch
Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models
pytorch resnet inception-resnet face-recognition face-identification face-tracking mtcnn pretrained-models face-detection facial-recognition
Language:Python 3367
THUDM / CodeGeeX
CodeGeeX: An Open Multilingual Code Generation Model
code-generation pretrained-models tools
Language:Python 3225
huawei-noah / Efficient-AI-Backbones
Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
convolutional-neural-networks efficient-inference ghostnet imagenet model-compression pretrained-models pytorch tensorflow transformer vision-transformer
Language:Python 3140
CLUEbenchmark / CLUE
中文语言理解测评基准 Chinese Language Understanding Evaluation Benchmark: datasets, baselines, pre-trained models, corpus and leaderboard
nlu benchmark chinese corpus dataset bert albert chineseglue glue roberta language-model pretrained-models transformers tensorflow pytorch
Language:Python 3107
osmr / imgclsmob
Sandbox for training deep learning networks
machine-learning deep-learning mxnet gluon pytorch classification imagenet neural-network image-classification chainer keras tensorflow pretrained-models cifar segmentation tensorflow2 semantic-segmentation human-pose-estimation 3d-face-reconstruction
Language:Python 2776
huawei-noah / Pretrained-Language-Model
Pretrained language model and its related optimization techniques developed by Huawei Noah's Ark Lab.
knowledge-distillation model-compression quantization pretrained-models large-scale-distributed
Language:Python 2654
PINTO_model_zoo
PINTO0309 / PINTO_model_zoo
A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.
tensorflow tensorflow-lite openvino edgetpu mediapipe coreml tensorflowjs tf-trt onnx pytorch tflite tfjs model-zoo models keras computer-vision caffe pretrained-models model tflite-models
Language:Python 2551
pyannote / pyannote-audio
Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech detection, speaker embedding
pytorch speech-processing speaker-diarization speech-activity-detection speaker-change-detection speaker-embedding voice-activity-detection pretrained-models overlapped-speech-detection speaker-recognition speaker-verification
Language:Python 2501
Separius / awesome-sentence-embedding
A curated list of pretrained sentence and word embedding models
wordembedding word-embeddings sentence-embeddings nlp pretrained-embedding awesome-list pretrained-models unsupervised-learning sentence-representations contextualized-representation awesome natural-language subword-models embedding-models cross-lingual bert pretrained-language-model language-model
Language:Python 2110
qubvel / efficientnet
Implementation of EfficientNet model. Keras and TensorFlow Keras.
classification imagenet efficientnet pretrained-models image-classification efficient nasnetmobile mobilenet deep-learning
Language:Python 2011
asteroid-team / asteroid
The PyTorch-based audio source separation toolkit for researchers
source-separation speech-separation audio-separation speech-enhancement deep-learning pytorch pretrained-models
Language:Python 1721
Fengshenbang-LM
IDEA-CCNL / Fengshenbang-LM
Fengshenbang-LM(封神榜大模型)是IDEA研究院认知计算与自然语言研究中心主导的大模型开源体系，成为中文AIGC和认知智能的基础设施。
aigc chinese-nlp distributed-training multimodal pretrained-models pytorch transformers
Language:Python 1702
OFA-Sys / OFA
Official repository of OFA (ICML 2022). Paper: OFA: Unifying Architectures, Tasks, and Modalities Through a Simple Sequence-to-Sequence Learning Framework
multimodal pretraining image-captioning text-to-image-synthesis visual-question-answering referring-expression-comprehension vision-language pretrained-models prompt prompt-tuning chinese
Language:Python 1660
FARM
deepset-ai / FARM
:house_with_garden: Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.
language-models bert nlp deep-learning transfer-learning pytorch nlp-library nlp-framework xlnet-pytorch ner question-answering pretrained-models roberta germanbert
Language:Python 1645
lonePatient / awesome-pretrained-chinese-nlp-models
Awesome Pretrained Chinese NLP Models，高质量中文预训练模型集合
chinese nlp pretrained-models bert albert roberta xlnet electra nezha ernie zen macbert gpt gpt-2 nlu-nlg t5-pegasus simbert nezha-gen pangu
1626
imcaspar / gpt2-ml
GPT2 for Multiple Languages, including pretrained models. GPT2 多语言支持, 15亿参数中文预训练模型
gpt-2 tpu bert pretrained-models chinese nlp tensorflow text-generation colab
Language:Python 1609
rwightman / gen-efficientnet-pytorch
Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS
mobilenetv3 mnasnet fbnet pytorch pretrained-models onnx caffe2 efficientnet
Language:Python 1518
EasyNLP
alibaba / EasyNLP
EasyNLP: A Comprehensive and Easy-to-use NLP Toolkit
transformers bert nlp pretrained-models deep-learning pytorch fewshot-learning knowledge-distillation knowledge-pretraining text-image-retrieval text-to-image-synthesis machine-learning text-classification transfer-learning
Language:Python 1495
deepsparse
neuralmagic / deepsparse
Inference runtime offering GPU-class performance on CPUs and APIs to integrate ML into your application
ml machinelearning pytorch tensorflow onnx deepsparse-engine sparsified-models sparsification-recipe inference computer-vision object-detection pruning quantization pretrained-models nlp auto-ml cpus yolov3 sparsification cpu-inference-api
Language:Python 1374
THUDM / CogView
Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".
text-to-image transformers pretrained-models pytorch
Language:Python 1312
Open3D-ML
isl-org / Open3D-ML
An extension of Open3D to address 3D Machine Learning tasks
3d-perception datasets pretrained-models lidar rgbd tensorflow pytorch visualization semantic-segmentation object-detection 3d-object-detection
Language:Python 1232
OFA-Sys / Chinese-CLIP
Chinese version of CLIP which achieves Chinese cross-modal retrieval and representation generation.
chinese computer-vision multi-modal-learning nlp pytorch vision-and-language-pre-training image-text-retrieval clip pretrained-models vision-language deep-learning multi-modal contrastive-loss transformers
Language:Python 1207
CV-pretrained-model
balavenkatesh3322 / CV-pretrained-model
A collection of computer vision pre-trained models.
computer-vision tensorflow keras pytorch mxnet models model pretrained-models pretrained data-science deep-learning neural-network awesome-list object-detection image-classification image video-analysis model-selection python3
1205
qubvel / classification_models
Classification models trained on ImageNet. Keras.
resnet pretrained-models pretrained-weights keras imagenet imagenet-classifier classification-model resnext senet squeeze-and-excitation inceptionv3 inception-resnet-v2 densenet mobilenet nasnet xception vgg
Language:Python 1163
kamalkraj / BERT-NER
Pytorch-Named-Entity-Recognition-with-BERT
bert named-entity-recognition pytorch conll-2003 cpp11 bert-ner inference curl postman pretrained-models
Language:Python 1114