SaaSHub helps you find the best software and product alternatives Learn more →
Top 20 Python pytorch-implementation Projects
-
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:
-
-
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.
-
-
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.
-
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.
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: https://github.com/thohemp/6DRepNet
-
I've still yet to try this one: https://github.com/huangzh13/StyleGAN.pytorch 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.
-
-
text-to-motion
Official implementation for "Generating Diverse and Natural 3D Human Motions from Texts."
-
explainable-cnn
📦 PyTorch based visualization package for generating layer-wise explanations for CNNs.
-
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 playground.py for learning more about OpenAI gym, etc.
-
-
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
-
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.
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: https://github.com/York-SDCNLab/Ghost-DeblurGAN, 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
Python pytorch-implementation related posts
- [D] Is it possible to train the same LLM instance on different users' data?
- How to create a custom parallel corpus for machine translation with recent versions of pytorch and torchtext?
- Help with VAE
- I need help knowing how to improve a CycleGan I am working on.
- Confusions on VAE implementation
- Confusion regarding Variational AutoEncoder Implementation
- Pytorch: Custom Dataset for Machine Translation
-
A note from our sponsor - #<SponsorshipServiceOld:0x00007f09211e1c08>
www.saashub.com | 7 Jun 2023
Index
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 |