Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

PyTorch

PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.

https://github.com/Siyu-C/ACAR-Net

[CVPR 2021] Actor-Context-Actor Relation Network for Spatio-temporal Action Localization

acar-net action-detection activitynet activitynet-challenge arxiv ava ava-dataset ava-kinetics model-zoo pytorch spatio-temporal-action-localization torch

Last synced: 03 Nov 2024

https://github.com/albumentations-team/autoalbument

AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - https://albumentations.ai/docs/autoalbument/

augmentation automated-machine-learning automl computer-vision deep-learning image-augmentation machine-learning pytorch

Last synced: 14 Feb 2025

https://github.com/jacobgil/pytorch-zssr

PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning

computer-vision deep-learning pytorch super-resolution

Last synced: 20 Dec 2024

https://github.com/jameschapman19/cca_zoo

Canonical Correlation Analysis Zoo: A collection of Regularized, Deep Learning based, Kernel, and Probabilistic methods in a scikit-learn style framework

canonical-correlation-analysis cca cca-zoo dcca deep kernel multiset-cca multiview pls pytorch tensor-cca

Last synced: 15 Feb 2025

https://ku-cvlab.github.io/Self-Attention-Guidance/

Official implementation of the paper "Improving Sample Quality of Diffusion Models Using Self-Attention Guidance" (ICCV 2023)

artificial-intelligence diffusion-models generative-models pytorch

Last synced: 31 Oct 2024

https://github.com/aws/sagemaker-pytorch-training-toolkit

Toolkit for running PyTorch training scripts on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers.

aws docker pytorch sagemaker

Last synced: 14 Feb 2025

https://github.com/p0p4k/pflowtts_pytorch

Unofficial implementation of NVIDIA P-Flow TTS paper

deep-learning flow-matching pflow pytorch text-to-speech tts

Last synced: 28 Nov 2024

https://github.com/matthew-z/R-net

R-net in PyTorch, with ELMo

allennlp pytorch r-net squad

Last synced: 02 Nov 2024

https://github.com/matthew-z/r-net

R-net in PyTorch, with ELMo

allennlp pytorch r-net squad

Last synced: 19 Dec 2024

https://github.com/catqaq/opentextclassification

OpenTextClassification is all you need for text classification! Open text classification for everyone, enjoy your NLP journey! 这可能是目前为止最全面的开源文本分类项目,支持中英双语、多种模型、多种任务。

bert bert-text-classification chinese-text-classification dpcnn ernie gru lstm multi-label-classification naive-bayes pytorch svm text-classification textcnn textrnn torchtext transformers

Last synced: 19 Feb 2025

https://github.com/yanndubs/neural-process-family

Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.

conditional-neural-process deep-learning machine-learning meta-learning neural-processes pytorch stochastic-processes uncertainty-estimation

Last synced: 13 Feb 2025

https://github.com/hangg7/deformable-kernels

Deforming kernels to adapt towards object deformation. In ICLR 2020.

iclr2020 operator pytorch representation-learning

Last synced: 20 Dec 2024

https://github.com/ikergarcia1996/easy-translate

Easy-Translate is a script for translating large text files with a SINGLE COMMAND. Easy-Translate is designed to be as easy as possible for beginners and as seamlesscustomizable and as possible for advanced users.

4-bit 8-bit begginers cpu easy easy-to-use gpu hugginface hugginface-hub huggingface-transformers llm m2m100 machine-translation nllb200 prompt pytorch quantization transformers translation

Last synced: 16 Feb 2025

https://github.com/andrewatanov/simclr-pytorch

PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results

contrastive-learning deep-learning pytorch pytorch-implementation representation-learning self-supervised-learning

Last synced: 17 Feb 2025

https://github.com/sml2h3/captcha_trainer_pytorch

基于MobileNetV2/EfficientNet-b0/... + LSTM + CTC的不定长图像识别训练pytorch框架

classfication efficientnet mobilenetv2 pytorch

Last synced: 27 Oct 2024

https://github.com/project-agml/agml

AgML is a centralized framework for agricultural machine learning. AgML provides access to public agricultural datasets for common agricultural deep learning tasks, with standard benchmarks and pretrained models, as well the ability to generate synthetic data and annotations.

agriculture computer-vision dataset deep-learning image-classification object-detection pytorch semantic-segmentation synthetic-data

Last synced: 19 Feb 2025

https://github.com/dusty-nv/jetson-voice

ASR/NLP/TTS deep learning inference library for NVIDIA Jetson using PyTorch and TensorRT

deep-learning jetson jetson-nano nlp pytorch speech-recognition tensorrt text-to-speech

Last synced: 18 Feb 2025

https://github.com/yerfor/syntaspeech

SyntaSpeech: Syntax-aware Generative Adversarial Text-to-Speech; IJCAI 2022; Official code

gan pytorch speech-synthesis tts

Last synced: 19 Dec 2024

https://github.com/kyegomez/visionmamba

Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model" It's 2.8x faster than DeiT and saves 86.8% GPU memory when performing batch inference to extract features on high-res images

ai machine-learning mamba pytorch recurrent-neural-network ssm

Last synced: 19 Feb 2025

https://github.com/learnables/cherry

A PyTorch Library for Reinforcement Learning Research

learning pytorch reinforcement reinforcement-learning rl

Last synced: 14 Feb 2025

https://github.com/tugstugi/pytorch-speech-commands

Speech commands recognition with PyTorch | Kaggle 10th place solution in TensorFlow Speech Recognition Challenge

cifar10 classification deep-learning densenet dual-path-networks kaggle neural-network pytorch resnet resnext speech-recognition wide-residual-networks

Last synced: 18 Feb 2025

https://github.com/OlgaChernytska/word2vec-pytorch

Implementation of the first paper on word2vec

deep-learning natural-language-processing pytorch word2vec

Last synced: 19 Jan 2025

https://github.com/shreyas-bk/u2netdemo

Demonstration using Google Colab to show how U-2-NET can be used for Background Removal, Changing Backgrounds, Bounding Box Creation, Salient Feature Highlighting and Salient Object Cropping.

background-removal bounding-boxes deep-learning image-cropping machine-learning python pytorch saliency-map tensorflow u2net

Last synced: 25 Oct 2024

https://github.com/coloquinte/torchsr

Super Resolution datasets and models in Pytorch

datasets edsr pytorch rcan rdn superresolution torchvision vdsr

Last synced: 19 Feb 2025

https://github.com/francescosaveriozuppichini/resnet

Clean, scalable and easy to use ResNet implementation in Pytorch

convolutional-neural-networks deep-learning neural-networks python3 pytorch

Last synced: 19 Dec 2024

https://github.com/rasbt/dora-from-scratch

LoRA and DoRA from Scratch Implementations

finetuning llm pytorch

Last synced: 19 Feb 2025

https://github.com/vt-vl-lab/guided-pix2pix

[ICCV 2019] Guided Image-to-Image Translation with Bi-Directional Feature Transformation

deep-learning gan pix2pix pytorch

Last synced: 19 Feb 2025

https://github.com/Karel911/TRACER

TRACER: Extreme Attention Guided Salient Object Tracing Network (AAAI 2022) implementation in PyTorch

aaai-2022 aaai2022 attention attention-mechanism background-removal image-segmentation pytorch pytorch-implementation salient-object-detection

Last synced: 02 Nov 2024

https://github.com/rlaphoenix/vsgan

PyTorch-based Super-Resolution and Restoration Image Processing Module for VapourSynth

deep-learning editing esrgan gan image-manipulation python pytorch restoration upscaling vapoursynth

Last synced: 17 Feb 2025

https://github.com/dccuchile/CC6204

Material del curso de Deep Learning de la Universidad de Chile

deep-learning pytorch uchile

Last synced: 27 Oct 2024

https://github.com/a-r-j/proteinworkshop

Benchmarking framework for protein representation learning. Includes a large number of pre-training and downstream task datasets, models and training/task utilities. (ICLR 2024)

benchmark dataset deep-learning lightning pretraining protein protein-structure pytorch

Last synced: 19 Feb 2025

https://github.com/wmcnally/golfdb

GolfDB is a video database for Golf Swing Sequencing, which involves detecting 8 golf swing events in trimmed golf swing videos. This repo demos the baseline model, SwingNet.

action-recognition event-detection golf pytorch temporal video

Last synced: 04 Dec 2024

https://github.com/minyoungg/pix2latent

Code for: Transforming and Projecting Images into Class-conditional Generative Networks

biggan cma-es gan inverting-images nevergrad pytorch stylegan2

Last synced: 04 Nov 2024

https://github.com/hityzy1122/opencv_transforms_torchvision

opencv reimplement for transforms in torchvision

opencv pillow pytorch torchvision

Last synced: 18 Feb 2025

https://github.com/LeapLabTHU/EfficientTrain

1.5−3.0× lossless training or pre-training speedup. An off-the-shelf, easy-to-implement algorithm for the efficient training of foundation visual backbones.

computer-vision deep-learning efficient-training machine-learning pytorch

Last synced: 05 Nov 2024

https://minyoungg.github.io/pix2latent/

Code for: Transforming and Projecting Images into Class-conditional Generative Networks

biggan cma-es gan inverting-images nevergrad pytorch stylegan2

Last synced: 11 Nov 2024

https://github.com/coqui-ai/trainer

🐸 - A general purpose model trainer, as flexible as it gets

ai data-science deep-learning machine-learning pytorch

Last synced: 17 Feb 2025

https://github.com/coqui-ai/Trainer

🐸 - A general purpose model trainer, as flexible as it gets

ai data-science deep-learning machine-learning pytorch

Last synced: 27 Nov 2024

https://github.com/eladrich/pix2vertex.pytorch

An official pyTorch port of the pix2vertex paper from ICCV2017

deep-learning iccv pytorch reconstruction

Last synced: 13 Feb 2025

https://github.com/csukuangfj/kaldifeat

Kaldi-compatible online & offline feature extraction with PyTorch, supporting CUDA, batch processing, chunk processing, and autograd - Provide C++ & Python API

cpp fbank features-extraction kaldi mfcc online-feature-extractor plp python pytorch streaming-feature-extractor

Last synced: 16 Feb 2025

https://github.com/openclimatefix/graph_weather

PyTorch implementation of Ryan Keisler's 2022 "Forecasting Global Weather with Graph Neural Networks" paper (https://arxiv.org/abs/2202.07575)

forecasting-models graph-neural-networks pytorch weather

Last synced: 15 Feb 2025

https://github.com/ai-trainee/traffic-sign-recognition-pyqt5-yolov5-gui

This YOLOv5🚀😊 GUI road sign system uses MySQL💽, PyQt5🎨, PyTorch, CSS🌈. It has modules for login🔑, YOLOv5 setup📋, sign recognition🔍, database💾, and image processing🖼️. It supports diverse inputs, model switching, and enhancements like mosaic and mixup📈.

css mysql-database pyqt5 pytorch tensorrt-engine yolov5 yolov5-gui

Last synced: 18 Feb 2025

https://github.com/chr5tphr/zennit

Zennit is a high-level framework in Python using PyTorch for explaining/exploring neural networks using attribution methods like LRP.

attribution deep-learning explainability explainable-ai feature-attribution interpretability interpretable-ai interpretable-ml lrp machine-learning python pytorch xai

Last synced: 30 Oct 2024

https://github.com/recursionpharma/gflownet

GFlowNet library specialized for graph & molecular data

deep-learning gflownet graph-neural-network pytorch

Last synced: 16 Nov 2024

https://github.com/Bobholamovic/CDLab

Yet another repository for developing and benchmarking deep learning-based change detection methods.

change-detection deep-learning pytorch remote-sensing

Last synced: 17 Nov 2024

https://github.com/locuslab/e2e-model-learning

Task-based end-to-end model learning in stochastic optimization

deep-learning machine-learning optimization paper pytorch stochastic-optimizers

Last synced: 10 Nov 2024

https://github.com/zjunlp/KnowPrompt

[WWW 2022] KnowPrompt: Knowledge-aware Prompt-tuning with Synergistic Optimization for Relation Extraction

dataset dialogre few-shot-learning knowledge-informed-prompt-learning knowprompt prompt prompt-learning prompt-tuning pytorch re relation-extraction semeval tacred text-classification

Last synced: 16 Nov 2024

https://github.com/yanndubs/hash-embeddings

PyTorch implementation of Hash Embeddings (NIPS 2017). Submission to the NIPS Implementation Challenge.

embeddings hashing nips nips-challenge nlp pytorch reproducible-research word-embeddings

Last synced: 27 Oct 2024

https://github.com/rayhane-mamah/efficient-vdvae

Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"

image-generation jax paper python pytorch representation-learning unsupervised-learning vae

Last synced: 19 Dec 2024

https://github.com/isl-org/Open3D-PointNet

Open3D PointNet implementation with PyTorch

open3d pointnet pytorch

Last synced: 28 Oct 2024

https://github.com/j3soon/tbparse

Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow

machine-learning parser plot python pytorch reader tensorboard tensorboardx tensorflow

Last synced: 16 Feb 2025

https://github.com/LambdaLabsML/distributed-training-guide

Best practices & guides on how to write distributed pytorch training code

cluster cuda deepspeed distributed-training fsdp gpu gpu-cluster kuberentes lambdalabs mpi nccl pytorch sharding slurm

Last synced: 21 Oct 2024

https://github.com/rlaphoenix/VSGAN

PyTorch-based Super-Resolution and Restoration Image Processing Module for VapourSynth

deep-learning editing esrgan gan image-manipulation python pytorch restoration upscaling vapoursynth

Last synced: 03 Nov 2024

https://github.com/divelab/good

GOOD: A Graph Out-of-Distribution Benchmark [NeurIPS 2022 Datasets and Benchmarks]

deep-learning distribution-shift graph-neural-networks graph-ood invariant-learning out-of-distribution-generalization pytorch pytorch-geometric

Last synced: 19 Feb 2025

https://github.com/ghimiredhikura/Complex-YOLOv3

PyTorch implementation of Complex-YOLO paper with YoloV3

deep-learning object-detection pytorch

Last synced: 09 Nov 2024

https://github.com/plantnet/plantnet-300k

[NeurIPS2021] A plant image dataset with high label ambiguity and a long-tailed distribution

dataset deep-learning plants pytorch

Last synced: 17 Feb 2025

https://github.com/yangyanli/do-conv

Depthwise Over-parameterized Convolutional Layer

convolutional-neural-networks deep-learning gluoncv pytorch tensorflow

Last synced: 18 Nov 2024

https://github.com/layumi/person-reid-triplet-loss

Person re-ID baseline with triplet loss

cnn image-retrieval open-reid personreid pytorch triplet-loss

Last synced: 20 Dec 2024

https://github.com/Randl/ShuffleNetV2-pytorch

Implementation of ShuffleNetV2 for pytorch

cnn-model pytorch shufflenetv2

Last synced: 31 Oct 2024

https://github.com/dansuh17/alexnet-pytorch

Pytorch Implementation of AlexNet

alexnet dataset paper pytorch

Last synced: 18 Feb 2025

https://github.com/guotong1988/NL2SQL-RULE

Content Enhanced BERT-based Text-to-SQL Generation https://arxiv.org/abs/1910.07179

bert deep-learning knowledge knowledge-representation nl2sql nlp pytorch rule-inject-to-model semantic-parsing text2sql

Last synced: 11 Nov 2024

https://github.com/lonepatient/lookahead_pytorch

pytorch implement of Lookahead Optimizer

cifar10 lookahead optimizer pytorch resnet-18

Last synced: 19 Dec 2024

https://github.com/lightforever/mlcomp

Distributed DAG (Directed acyclic graph) framework for machine learning with UI

artificial-intelligence automl computer-vision deep-learning distributed-computing infrastructure machine-learning python pytorch research

Last synced: 19 Dec 2024

https://github.com/cmsflash/beauty-net

A simple, flexible, and extensible template for PyTorch. It's beautiful.

cnn computer-vision deep-learning image-classification pytorch pytorch-template

Last synced: 10 Nov 2024

https://github.com/flyingpot/pytorch_deephash

Pytorch implementation of Deep Learning of Binary Hash Codes for Fast Image Retrieval, CVPRW 2015

deep-learning deephash hash image-retrieval pytorch

Last synced: 19 Dec 2024

https://github.com/fanglanting/skip-gram-pytorch

A complete pytorch implementation of skip-gram

embed pytorch spearman word2vec

Last synced: 27 Nov 2024

https://github.com/luopeixiang/im2latex

Pytorch implemention of Deep CNN Encoder + LSTM Decoder with Attention for Image to Latex

encoder-decoder-model im2latex imagecaptioning pytorch seq2seq show-and-tell

Last synced: 24 Jan 2025

https://github.com/dair-ai/emotion_dataset

:smile: Dataset for Emotion Recognition Research

dataset machine-learning nlp pytorch

Last synced: 27 Dec 2024

https://github.com/liuzuxin/osrl

🤖 Elegant implementations of offline safe RL algorithms in PyTorch

bc-safe cdt cpq library offline-rl offline-safe-rl pytorch reinforcement-learning robotics safe-rl

Last synced: 15 Feb 2025

https://github.com/nianticlabs/depth-hints

[ICCV 2019] Depth Hints are complementary depth suggestions which improve monocular depth estimation algorithms trained from stereo pairs

computer-vision deep-learning depth-estimation monodepth neural-network pytorch self-supervision stereo-matching

Last synced: 20 Dec 2024

https://github.com/Project-AgML/AgML

AgML is a centralized framework for agricultural machine learning. AgML provides access to public agricultural datasets for common agricultural deep learning tasks, with standard benchmarks and pretrained models, as well the ability to generate synthetic data and annotations.

agriculture computer-vision dataset deep-learning image-classification object-detection pytorch semantic-segmentation synthetic-data

Last synced: 14 Nov 2024