#Pytorch-implementation
Showing 60 of 78 repositories tagged #pytorch-implementation, ranked by stars
Translate manga/image 一键翻译各类图片内文字 https://cotrans.touhou.ai/ (no longer working)
A resource for learning about Machine learning & Deep Learning
A Collection of Variational Autoencoders (VAE) in PyTorch.
🧠💬 Articles I wrote about machine learning, archived from MachineCurve.com.
[CVPR2020] Adversarial Latent Autoencoders
🔎 🖼️ 🔥PyTorch Toolbox for Image Quality Assessment, including PSNR, SSIM, LPIPS, FID, NIQE, NRQM(Ma), MUSIQ, TOPIQ, NIMA, DBCNN, BRISQUE, PI and more...
My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!
PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
This repository is the official implementation of Disentangling Writer and Character Styles for Handwriting Generation (CVPR 2023)
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.
State-of-the-art, simple, fast unbounded / large-scale NeRFs.
A simple and extensible library to create Bayesian Neural Network layers on PyTorch.
A machine learning library with a TypeScript API and Rust backend. CUDA and WebGPU compatibility. Built to understand how ML frameworks and models work internally.
[CVPR 2022] FaceFormer: Speech-Driven 3D Facial Animation with Transformers
Pytorch implementation for "Large-Scale Long-Tailed Recognition in an Open World" (CVPR 2019 ORAL)
From scratch implementation of a sparse mixture of experts language model inspired by Andrej Karpathy's makemore :)
A Pytorch Implementation of "Neural Speech Synthesis with Transformer Network"
Benchmark of federated learning. Dedicated to the community. 🤗
Build high-performance AI models with modular building blocks
Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.
Photometric optimization code for creating the FLAME texture space and other applications
Voice Conversion by CycleGAN (语音克隆/语音转换): CycleGAN-VC2
Official Code for ECCV 2024 paper — One-Shot Diffusion Mimicker for Handwritten Text Generation
The official PyTorch implementation of L2CS-Net for gaze estimation and tracking
The full collection of all codes for my Youtube Channel segregated as per topic.
This is an open sourced book on deep learning.
Pytorch Tutorial, Pytorch with Google Colab, Pytorch Implementations: CNN, RNN, DCGAN, Transfer Learning, Chatbot, Pytorch Sample Codes
Generating RGB photos from RAW image files with PyNET (PyTorch)
The official implementation of CFBI(+): Collaborative Video Object Segmentation by (Multi-scale) Foreground-Background Integration.
Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).
A PyTorch-based Semi-Supervised Learning (SSL) Codebase for Pixel-wise (Pixel) Vision Tasks [ECCV 2020]
推荐系统论文算法实现，包括序列推荐，多任务学习，元学习等。 Recommendation system papers implementations, including sequence recommendation, multi-task learning, meta-learning, etc.
From scratch implementation of a vision language model in pure PyTorch
This repo is official PyTorch implementation of MobileHumanPose: Toward real-time 3D human pose estimation in mobile devices(CVPRW 2021).
[CVPR2023] Blur Interpolation Transformer for Real-World Motion from Blur
An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.
📦 PyTorch based visualization package for generating layer-wise explanations for CNNs.
Pretrained GANs + VAEs + classifiers for MNIST/CIFAR in pytorch.
Online Deep Learning: Learning Deep Neural Networks on the Fly / Non-linear Contextual Bandit Algorithm (ONN_THS)
Detailed python notes & code for lectures and exercises of Andrej Karpathy's course "Neural Networks: Zero to Hero." The course is focused on building neural networks from scratch.
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.
A tool for precisely placing 3D landmarks on 3D facial scans based on the paper "Multi-view Consensus CNN for 3D Facial Landmark Placement"
This repository contains the collection of explorative notebooks pure in python and in the language that we, humans can read. Have tried to compile all lectures from the Andrej Karpathy's 💎 playlist on Neural Networks - which we will end up with building GPT.
PyTorch Implementation of Deep Hierarchical Classification for Category Prediction in E-commerce System
Implementation of Resnet-50 with and without CBAM in PyTorch v1.8. Implementation tested on Intel Image Classification dataset from https://www.kaggle.com/puneet6060/intel-image-classification.
This is an unofficial reproduction of paper Moir´e Photo Restoration Using Multiresolution Convolutional Neural Networks.(PyTorch)
Implementation of soft dynamic time warping in pytorch
Pytorch implementation of Paper by Google Research - Representation Learning for Information Extraction from Form-like Documents.
Pytorch implementation of Unsupervised Attention-guided Image-to-Image Translation.
PyTorch implementation of Looking Fast and Slow: Memory-Guided Mobile Video Object Detection
Instance Segmentation for Chinese Character Stroke Extraction, Datasets and Benchmarks.
This repository guides you through the process of building a GPT-style Large Language Model (LLM) from scratch using PyTorch. The structure and approach are inspired by the book Build a Large Language Model (From Scratch) by Sebastian Raschka.
Official PyTorch implementation of "Deep Slow Motion Video Reconstruction with Hybrid Imaging System" (TPAMI)
Absolutely amazing SOTA Google Colab (Jupyter) Notebooks for creating/training SOTA Music AI models and for generating music with Transformer technology (Google XLNet/Transformer-XL)
[ECCV 2022] Official repository of DeMFI.
A PyTorch Implementation of "Optimization of Molecules via Deep Reinforcement Learning".
Pytorch implementation of "Learning Lightweight Lane Detection CNNs by Self Attention Distillation (ICCV 2019)"
Test-Time Augmentation library for Pytorch
Conformal prediction for controlling monotonic risk functions. Simple accompanying PyTorch code for conformal risk control in computer vision and natural language processing.
PyTorch implementation of COVID-Net https://github.com/lindawangg/COVID-Net