Python pytorch-implementation

Open-source Python projects categorized as pytorch-implementation

Top 20 Python pytorch-implementation Projects

  • Machine-Learning-Collection

    A resource for learning about Machine learning & Deep Learning

    Project mention: How to create a custom parallel corpus for machine translation with recent versions of pytorch and torchtext? | /r/pytorch | 2023-02-19

    I am trying to train a model for NMT on a custom dataset. I found this great tutorial on youtube along with the accompanying repo, but it uses an old version of PyTorch and torchtext. More recent versions of torchtext have removed the Field and BucketIterator classes. I looked for more recent tutorials. The closest thing I could find was this medium post (again with the accompanying code) which worked with a custom dataset for text classification. I tried to replicate the code with my problem and got this far:

  • PyTorch-VAE

    A Collection of Variational Autoencoders (VAE) in PyTorch.

    Project mention: Help with VAE | /r/MLQuestions | 2022-11-28
  • Sonar

    Write Clean Python Code. Always.. Sonar helps you commit clean code every time. With over 225 unique rules to find Python bugs, code smells & vulnerabilities, Sonar finds the issues while you focus on the work.

  • ALAE

    [CVPR2020] Adversarial Latent Autoencoders

  • medicaldetectiontoolkit

    The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

  • manga-image-translator

    Translate manga/image 一键翻译各类图片内文字

    Project mention: Do you guys know where I can read the translated version of Isekai Joshi Kangoku? | /r/shoujoai | 2023-05-16
  • 6DRepNet

    Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.

    Project mention: How to do Human Head Segmentation from images? | /r/computervision | 2023-03-27

    Well so it sounds like you want to do something like DeepFaceLab? I don't have a lot of experience with that but I believe they collect facial images based on their angles (pitch, roll and yaw) and match and merge them together. So something like this might help you there:

  • StyleGAN.pytorch

    A PyTorch implementation for StyleGAN with full features.

    Project mention: I'm stumped with installing PyTorch. | /r/pytorch | 2022-10-04

    I've still yet to try this one: but at this point it might just be worth trying the Jupiter thingie. I sort of understand what it is.

  • InfluxDB

    Access the most powerful time series database as a service. Ingest, store, & analyze all types of time series data in a fully-managed, purpose-built database. Keep data forever with low-cost storage and superior data compression.

  • ConvLSTM-PyTorch

    ConvLSTM/ConvGRU (Encoder-Decoder) with PyTorch on Moving-MNIST

  • text-to-motion

    Official implementation for "Generating Diverse and Natural 3D Human Motions from Texts."

    Project mention: AI based Text to Motion | /r/godot | 2023-01-30
  • explainable-cnn

    📦 PyTorch based visualization package for generating layer-wise explanations for CNNs.

  • ElasticFace

    Official repository for ElasticFace: Elastic Margin Loss for Deep Face Recognition

    Project mention: Face recognition models require different thresholds for different races? [D] | /r/MachineLearning | 2023-05-26

    Hey, first of all, don't use FaceNet. It is outdated and performs way worse than modern solutions. You can use Elastic Face or MagFace (or any other SOTA FR model).

  • pytorch-learn-reinforcement-learning

    A collection of various RL algorithms like policy gradients, DQN and PPO. The goal of this repo will be to make it a go-to resource for learning about RL. How to visualize, debug and solve RL problems. I've additionally included for learning more about OpenAI gym, etc.

  • torch-metrics

    Metrics for model evaluation in pytorch

  • Data-Efficient-Reinforcement-Learning-with-Probabilistic-Model-Predictive-Control

    Unofficial Implementation of the paper "Data-Efficient Reinforcement Learning with Probabilistic Model Predictive Control", applied to gym environments

  • Pix2Seq

    Simple Implementation of Pix2Seq model for object detection in PyTorch

    Project mention: [P] Simple Implementation of Pix2Seq in PyTorch - Object Detection w/ Transformers | /r/MachineLearning | 2022-08-30

    Link to my Towards AI blog Link to the project on my GitHub. Link to the Colab Notebook

  • active_learning

    Code for Active Learning at The ImageNet Scale. This repository implements many popular active learning algorithms and allows training with torch's DDP.

  • Ghost-DeblurGAN

    This is a lightweight GAN developed for real-time deblurring. The model has a super tiny size and a rapid inference time. The motivation is to boost marker detection in robotic applications, however, you may use it for other applications definitely.

    Project mention: Apriltag pose detection on curved surface | /r/robotics | 2022-08-03

    You could try some sort of Image restoration GAN to produce a modified output of the input image (which lies on the curved surface). I have conducted research on GANs for deblurring Fiducial markers, and simply put, the Network takes a blurred image containing fiducial markers and outputs a ‘deblurred image’ whose fiducial markers can be more easily detected. However, keep in mind that the GANs can perform a wide variety of operations on an input Image (other than just deblurring), hence, they are called Image restoration networks in general. What you could give a shot is, create a pipeline where the captured image goes into the GAN, then the output is fed to the Apriltag detector. Here is the link to our GitHub if you need a starter:, and some of my other suggestions are NAFNet and HINet.

  • 3d-multi-resolution-rcnn

    Official PyTorch implementaiton of the paper "3D Instance Segmentation Framework for Cerebral Microbleeds using 3D Multi-Resolution R-CNN."

  • Amortized-SVGD-GAN

    Learning to draw samples: with application to amortized maximum likelihood estimator for generative adversarial learning

  • Robo-Semantic-Segmentation

    Just a simple semantic segmentation library that I developed to speed up the image segmentation pipeline

  • SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

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 2023-05-26.

Python pytorch-implementation related posts


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

Project Stars
1 Machine-Learning-Collection 5,359
2 PyTorch-VAE 4,822
3 ALAE 3,399
4 medicaldetectiontoolkit 1,204
5 manga-image-translator 1,109
6 6DRepNet 356
7 StyleGAN.pytorch 313
8 ConvLSTM-PyTorch 279
9 text-to-motion 242
10 explainable-cnn 195
11 ElasticFace 132
12 pytorch-learn-reinforcement-learning 130
13 torch-metrics 108
14 Data-Efficient-Reinforcement-Learning-with-Probabilistic-Model-Predictive-Control 76
15 Pix2Seq 71
16 active_learning 51
17 Ghost-DeblurGAN 24
18 3d-multi-resolution-rcnn 19
19 Amortized-SVGD-GAN 8
20 Robo-Semantic-Segmentation 0
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives