Python pretrained-models

Open-source Python projects categorized as pretrained-models | Edit details

Top 23 Python pretrained-model Projects

  • GitHub repo transformers

    🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

    Project mention: Top Github repo trends in 2021 | | 2022-01-12

    No surprises here: deep learning is the most popular subcategory, with hugging face transformers repo, YOLOv5, Tensorflow and Deepmind’s Alphafold all in the mix. Surprisingly, the only proper infrastructure-ey repos on the list are Meilisearch and Clickhouse, a tad bit surprising given all the hype data infrastructure receives in VC-world, but again, probably just a question of size of end-user populations + whether data scientists spend tons of time on Github vs. Web Developers…

  • GitHub repo spleeter

    Deezer source separation library including pretrained models.

    Project mention: Dissecting sun seeping through (from pocket mirror) | | 2022-01-19

    All of these apps use the Deezer algorithm anyway. You don't need an app for though this likely works better on desktop.

  • SonarQube

    Static code analysis for 29 languages.. Your projects are multi-language. So is SonarQube analysis. Find Bugs, Vulnerabilities, Security Hotspots, and Code Smells so you can release quality code every time. Get started analyzing your projects today for free.

  • GitHub repo nlp-recipes

    Natural Language Processing Best Practices & Examples

    Project mention: Is there any utility software/bot that produces descriptor tags for a Reddit image post using the comments? | | 2021-11-07

    I found this ( resource and it has a list of pre-built or easily customizable NLP models that I'm going to try out.

  • GitHub repo segmentation_models.pytorch

    Segmentation models with pretrained backbones. PyTorch.

    Project mention: Advice needed | | 2021-10-02

    You could also use qubvel's segmentation models if you would like to explore semantic segmentation.

  • GitHub repo PaddleClas

    A treasure chest for visual recognition powered by PaddlePaddle

    Project mention: Baidu AI Research Team Introduces ‘PP-ShiTu’: A Practical Lightweight Image Recognition System | | 2021-11-07

    Quick 5 Min Read | Paper | Github

  • GitHub repo facenet-pytorch

    Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models

    Project mention: Query regarding Multiple face recognization system | | 2021-11-25

    It's generally better to split the task into a multiple tasks. First I'd want to detect and extract faces. There are a number of pretrained models that you could use for that, e.g., Once you've extracted faces, you can train a facial recognition using something like a siamese network as you normally would.

  • GitHub repo efficientnet

    Implementation of EfficientNet model. Keras and TensorFlow Keras.

    Project mention: I made an image recognition model written in NodeJs | | 2021-02-24

    EfficientNet a lightweight convolutional neural network architecture achieving the state-of-the-art accuracy with an order of magnitude fewer parameters and FLOPS, on both ImageNet and five other commonly used transfer learning datasets.

  • OPS

    OPS - Build and Run Open Source Unikernels. Quickly and easily build and deploy open source unikernels in tens of seconds. Deploy in any language to any cloud.

  • GitHub repo FARM

    :house_with_garden: Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.

    Project mention: Can someone please explain to me the differences between train, dev and test datasets? | | 2021-09-16

    I'm also trying to solve this task in a python notebook (.ipynb) using the FARM framework and BERT model of huggingface

  • GitHub repo PINTO_model_zoo

    A repository that shares tuning results of trained models generated by TensorFlow / Keras. Post-training quantization (Weight Quantization, Integer Quantization, Full Integer Quantization, Float16 Quantization), Quantization-aware training. TensorFlow Lite. OpenVINO. CoreML. TensorFlow.js. TF-TRT. MediaPipe. ONNX. [.tflite,.h5,.pb,saved_model,tfjs,tftrt,mlmodel,.xml/.bin, .onnx]

    Project mention: Loading Saved Models for transfer learning | | 2021-08-02

    Check it out

  • GitHub repo pyannote-audio

    Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech detection, speaker embedding

    Project mention: [D] What is the best package for combined speech recognition and diarization on long conversation audio files? | | 2021-06-13

    What about ?

  • GitHub repo asteroid

    The PyTorch-based audio source separation toolkit for researchers

    Project mention: Speech separation | | 2021-04-12
  • GitHub repo BERT-NER


    Project mention: Training NER models for detecting custom entities | | 2021-10-08
  • GitHub repo trt_pose

    Real-time pose estimation accelerated with NVIDIA TensorRT

    Project mention: How to setup gesture recognition on the Nano? | | 2021-04-21

    I just followed the setup instructions for TRT_pose and then TRT_pose_hand and it was working great for me! You will need to follow these steps first to get TRT_pose on your device.

  • GitHub repo PatrickStar

    PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.

    Project mention: [R] Patrickstar: A Pytorch Based Training Framework That Can Train Faster And Larger PTMs | | 2021-11-29

    As large PTM are becoming a must in NLP, we believe PatrickStar would be a help to researchers and engineers. It would be so nice if you could checkout our github repo and give us some feedback!

  • GitHub repo virtex

    [CVPR 2021] VirTex: Learning Visual Representations from Textual Annotations

    Project mention: [D] Good quality code repos on deep learning | | 2021-08-27
  • GitHub repo finetuner

    Finetuning any DNN for better embedding on neural search tasks

    Project mention: The hand-picked selection of the best Python libraries released in 2021 | | 2021-12-21

    Jina and Finetuner.

  • GitHub repo deepsparse

    Neural network inference engine that delivers GPU-class performance for sparsified models on CPUs

    Project mention: [D] How does the DeepSparse Engine work? | | 2021-10-23

    Neural magic has open source tools for pruning models and the DeepSparse efficiently executes the sparse models on CPU. Here is the only description of the inner workings I could find. source

  • GitHub repo huggingface_hub

    All the open source things related to the Hugging Face Hub.

    Project mention: Machine Learning for Developers & PMs | | 2022-01-19

    Hey, I’m Merve from Hugging Face, an Open-Source Company working in the democratization of responsible Machine Learning. 👋

  • GitHub repo diffwave

    DiffWave is a fast, high-quality neural vocoder and waveform synthesizer.

    Project mention: Interesting research areas in Digital Signal Processing. | | 2021-06-01

    Here’s another one

  • GitHub repo wavegrad

    A fast, high-quality neural vocoder.

    Project mention: WaveGrad implementation and pretrained model | | 2021-05-31
  • GitHub repo flaxmodels

    Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

    Project mention: [P] Training StyleGAN2 in Jax (FFHQ and Anime Faces) | | 2021-09-12

    a while ago I posted here regarding a repository containing some pretrained models implemented in Jax/Flax. I decided to add training code for these models. Here is the training code for StyleGAN2:

  • GitHub repo opti_models

    PyTorch optimizations and benchmarking

    Project mention: How to get from zero to production ready neural nets? | | 2021-07-17

    There are many ways to convert models into different formats for production: ONNX, OPENVINO, TensorRT. Here is a repo with examples:

  • GitHub repo mammography_metarepository

    Meta-repository of screening mammography classifiers

    Project mention: [R] Meta-repository of mammography classifiers | | 2021-08-11

    Code for found:

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020). The latest post mention was on 2022-01-19.

Python pretrained-models related posts


What are some of the best open-source pretrained-model projects in Python? This list will help you:

Project Stars
1 transformers 57,096
2 spleeter 18,487
3 nlp-recipes 5,798
4 segmentation_models.pytorch 4,752
5 PaddleClas 3,230
6 facenet-pytorch 2,622
7 efficientnet 1,932
8 FARM 1,451
9 PINTO_model_zoo 1,382
10 pyannote-audio 1,363
11 asteroid 1,242
12 BERT-NER 990
13 trt_pose 679
14 PatrickStar 542
15 virtex 490
16 finetuner 402
17 deepsparse 363
18 huggingface_hub 316
19 diffwave 264
20 wavegrad 169
21 flaxmodels 82
22 opti_models 59
23 mammography_metarepository 29
Find remote jobs at our new job board There are 29 new remote jobs listed recently.
Are you hiring? Post a new remote job listing for free.
Less time debugging, more time building
Scout APM allows you to find and fix performance issues with no hassle. Now with error monitoring and external services monitoring, Scout is a developer's best friend when it comes to application development.