PyTorch | Ecosyste.ms: Awesome

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: 16 Nov 2024

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/Bil369/MaskDetect-YOLOv4-PyTorch

基于PyTorch&YOLOv4实现的口罩佩戴检测 :star: 自建口罩数据集分享

maskdetect pytorch yolov4

Last synced: 09 Nov 2024

https://github.com/bytedance/flux

A fast communication-overlapping library for tensor parallelism on GPUs.

cuda cutlass gpu pytorch

Last synced: 15 Nov 2024

https://github.com/curiousily/deploy-bert-for-sentiment-analysis-with-fastapi

Deploy BERT for Sentiment Analysis as REST API using FastAPI, Transformers by Hugging Face and PyTorch

bert deep-learning deploy-machine-learning deployment fastapi huggingface huggingface-transformer machine-learning python pytorch rest rest-api sentiment-analysis transformers uvicorn

Last synced: 18 Nov 2024

https://github.com/islanna/dynamicrelu

Implementation of Dynamic ReLU on Pytorch

dynamic pytorch relu

Last synced: 15 Nov 2024

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/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: 31 Oct 2024

https://github.com/graphsignal/graphsignal-python

Graphsignal Tracer for Python

ai-agents artificial-intelligence debugging deep-learning huggingface inference langchain langchain-python llama-index machine-learning monitoring observability openai-api python pytorch tracer

Last synced: 11 Oct 2024

https://github.com/ssube/onnx-web

web UI for GPU-accelerated ONNX pipelines like Stable Diffusion, even on Windows and AMD

ai-art amd diffusion esrgan flask generative-art gfpgan image-generation linux nvidia python pytorch reactjs stable-diffusion super-resolution text2image upscaling web-app windows

Last synced: 14 Oct 2024

https://github.com/tomchang25/whisper-auto-transcribe

Auto transcribe tool based on whisper

asr deep-learning gradio gradio-interface language-model pytorch speech-processing speech-recognition speech-to-text text-to-speech video-captioning voice-activity-detection

Last synced: 04 Aug 2024

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: 15 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/hangg7/deformable-kernels

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

iclr2020 operator pytorch representation-learning

Last synced: 16 Nov 2024

https://github.com/hyperdashio/hyperdash-sdk-py

Official Python SDK for Hyperdash

ai ipython jupyter-notebook keras machine-learning pytorch sdk

Last synced: 12 Oct 2024

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

R-net in PyTorch, with ELMo

allennlp pytorch r-net squad

Last synced: 14 Oct 2024

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/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: 16 Nov 2024

https://github.com/sony/bigvsan

Pytorch implementation of BigVSAN

audio-generation audio-synthesis gan neural-vocoder pytorch speech-synthesis

Last synced: 15 Nov 2024

https://github.com/sudharsan13296/getting-started-with-google-bert

Build and train state-of-the-art natural language processing models using BERT

albert bart bert bertsum clinical-bert distilbert electra huggingface-transformers nlp pytorch roberta sbert sentence-bert spanbert tinybert transformer videobert

Last synced: 15 Nov 2024

https://github.com/napsternxg/pytorch-practice

Some example scripts on pytorch

deep-learning machine-learning pytorch pytorch-tutorial

Last synced: 18 Nov 2024

https://github.com/sea-snell/implicit-language-q-learning

Official code from the paper "Offline RL for Natural Language Generation with Implicit Language Q Learning"

implicit-q-learning iql language-model nlp offline-rl python pytorch q-learning reinforcement-learning

Last synced: 18 Nov 2024

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/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: 07 Oct 2024

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: 15 Nov 2024

https://github.com/opendilab/di-hpc

OpenDILab RL HPC OP Lib

cuda hpc lstm pytorch reinforcement-learning

Last synced: 09 Nov 2024

https://github.com/Peterisfar/YOLOV3

yolov3 by pytorch

mobilenetv2 model-compression object-detection pytorch voc yolov3

Last synced: 09 Nov 2024

https://github.com/ENSTA-U2IS-AI/torch-uncertainty

Open-source framework for uncertainty and deep learning models in PyTorch :seedling:

bayesian-network computer-vision ensembles mixup neural-networks predictive-uncertainty pytorch reliable-ai trustworthy-machine-learning uncertainty uncertainty-quantification

Last synced: 05 Nov 2024

https://github.com/qanastek/HugsVision

HugsVision is a easy to use huggingface wrapper for state-of-the-art computer vision

bert computer-vision deep-learning deit detr huggingface image-classification image-generation machine-learning object-detection pretrained-models pythorch pytorch pytorch-transformers semantic-segmentation state-of-the-art torchvision transformers vit yolo

Last synced: 15 Nov 2024

https://github.com/szymonmaszke/torchfunc

PyTorch functions and utilities to make your life easier

docker extensions functions neural-network performance performance-analysis pytorch record recording tips utilities utils

Last synced: 16 Nov 2024

https://github.com/qanastek/hugsvision

HugsVision is a easy to use huggingface wrapper for state-of-the-art computer vision

bert computer-vision deep-learning deit detr huggingface image-classification image-generation machine-learning object-detection pretrained-models pythorch pytorch pytorch-transformers semantic-segmentation state-of-the-art torchvision transformers vit yolo

Last synced: 16 Nov 2024

https://github.com/AnshMittal1811/MachineLearning-AI

This repository contains all the work that I regularly did and studied from Medium blogs, several research papers, and other Repos (related/unrelated to the research papers).

3d-computer-vision audio-signal-processing computer-vision convolutional-neural-networks deep-learning deep-neural-networks generative-models gradcam graph-neural-networks image-classification lidar-point-cloud machine-learning neural-network neural-networks neural-radiance-fields neural-rendering pytorch transformers vision-transformer

Last synced: 28 Oct 2024

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/google/jetstream

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

gemma gpt gpu inference jax large-language-models llama llama2 llm llm-inference llmops mlops model-serving pytorch tpu transformer

Last synced: 29 Sep 2024

https://github.com/r00tman/NeRF-OSR

NeRF for Outdoor Scene Relighting [ECCV 2022]

computer-vision dataset eccv2022 illumination mesh nerf neural-radiance-fields neural-rendering openvr pytorch reflectance relighting volume-rendering vr

Last synced: 07 Nov 2024

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: 08 Aug 2024

https://github.com/Oulu-IMEDS/pytorch_bn_fusion

Batch normalization fusion for PyTorch

batch-normalization deep-learning deep-neural-networks inference-optimization pytorch

Last synced: 07 Aug 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/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/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: 16 Nov 2024

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/coqui-ai/trainer

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

ai data-science deep-learning machine-learning pytorch

Last synced: 15 Nov 2024

https://github.com/hityzy1122/opencv_transforms_torchvision

opencv reimplement for transforms in torchvision

opencv pillow pytorch torchvision

Last synced: 12 Nov 2024

https://github.com/Xharlie/BtcDet

Behind the Curtain: Learning Occluded Shapes for 3D Object Detection

3d 3d-object-detection aaai-2022 aaai2022 autonomous-driving autonomous-vehicles computer-vision deep-learning generative-model lidar lidar-point-cloud object-detection occlusion occlusion-handling point-cloud pytorch self-driving self-driving-car

Last synced: 28 Oct 2024

https://github.com/pycroscopy/atomai

Deep and Machine Learning for Microscopy

deep-kernel-learning deep-learning electron-microscopy ensemble-learning fully-convolutional-networks google-colaboratory imaging jupyter-notebook machine-learning materials materials-science microscopy multivariate-analysis pytorch scanning-probe-microscopy semantic-segmentation variational-autoencoders

Last synced: 14 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 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/francescosaveriozuppichini/resnet

Clean, scalable and easy to use ResNet implementation in Pytorch

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

Last synced: 15 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/eladrich/pix2vertex.pytorch

An official pyTorch port of the pix2vertex paper from ICCV2017

deep-learning iccv pytorch reconstruction

Last synced: 15 Nov 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: 14 Nov 2024

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/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: 14 Nov 2024

https://github.com/taited/clip-score

Quick scripts to calculate CLIP text-image similarity

batch clip clip-score pytorch

Last synced: 13 Nov 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/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: 13 Aug 2024

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

Open3D PointNet implementation with PyTorch

open3d pointnet pytorch

Last synced: 28 Oct 2024

https://github.com/pythonlessons/mltu

Machine Learning Training Utilities (for TensorFlow and PyTorch)

machine-learning object-detection ocr pytorch speech-recognition speech-to-text tensorflow yolov8

Last synced: 13 Nov 2024

https://github.com/lettucecfd/lettuce

Computational Fluid Dynamics based on PyTorch and the Lattice Boltzmann Method

cfd lattice-boltzmann machine-learning physics-simulation pytorch torch

Last synced: 05 Aug 2024

https://github.com/lquirosd/P2PaLA

Page to PAGE Layout Analysis Tool

computer-vision deep-neural-networks document-layout-analysis gan generative-adversarial-network handwritten-text-recognition image-segmentation page-xml pix2pix pytorch

Last synced: 03 Nov 2024

https://github.com/greydanus/mnist1d

A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.

convnet dataset machine-learning neural-networks pytorch research

Last synced: 01 Oct 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/autonise/CRAFT-Remade

Implementation of CRAFT Text Detection

craft detection ocr pytorch pytorch-implementation text-detection weak-supervision

Last synced: 03 Nov 2024

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/curiousily/Deploy-BERT-for-Sentiment-Analysis-with-FastAPI

Deploy BERT for Sentiment Analysis as REST API using FastAPI, Transformers by Hugging Face and PyTorch

bert deep-learning deploy-machine-learning deployment fastapi huggingface huggingface-transformer machine-learning python pytorch rest rest-api sentiment-analysis transformers uvicorn

Last synced: 13 Aug 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/ghimiredhikura/Complex-YOLOv3

PyTorch implementation of Complex-YOLO paper with YoloV3

deep-learning object-detection pytorch

Last synced: 09 Nov 2024

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

Implementation of ShuffleNetV2 for pytorch

cnn-model pytorch shufflenetv2

Last synced: 31 Oct 2024

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/OlgaChernytska/word2vec-pytorch

Implementation of the first paper on word2vec

deep-learning natural-language-processing pytorch word2vec

Last synced: 26 Sep 2024

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/clovaai/fewshot-font-generation

The unified repository for few-shot font generation methods. This repository includes FUNIT (ICCV'19), DM-Font (ECCV'20), LF-Font (AAAI'21) and MX-Font (ICCV'21).

deep-learning fewshot-font-generation font font-generation generative-models machine-learning pytorch

Last synced: 12 Nov 2024

https://github.com/nicklashansen/voice-activity-detection

Voice Activity Detection (VAD) using deep learning.

convolutional-neural-networks deep-learning deep-neural-networks densenet focal-loss pytorch recurrent-neural-networks voice-activity-detection

Last synced: 14 Nov 2024

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

LoRA and DoRA from Scratch Implementations

finetuning llm pytorch

Last synced: 15 Nov 2024

https://github.com/eric-canas/qreader

Robust and Straight-Forward solution for reading difficult and tricky QR codes within images in Python. Powered by YOLOv8

computer-vision easy-to-use image-processing object-detection pip python pytorch pyzbar qr qrcode qrcode-reader qrcode-scanner yolov8

Last synced: 13 Nov 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: 15 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: 16 Nov 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/dair-ai/emotion_dataset

:smile: Dataset for Emotion Recognition Research

dataset machine-learning nlp pytorch

Last synced: 17 Nov 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: 17 Nov 2024

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: 22 Oct 2024

https://github.com/pengboxiangshang/torchsketch

TPAMI 2022

cnn deep-learning gnn pytorch rnn sketch tcn

Last synced: 10 Oct 2024

https://github.com/li-plus/dsnet

DSNet: A Flexible Detect-to-Summarize Network for Video Summarization

computer-vision detection machine-learning pytorch video-summarization

Last synced: 06 Nov 2024

https://github.com/blue-season/pywarm

A cleaner way to build neural networks for PyTorch.

clean-code data-science deep-learning keras machine-learning neural-network neural-networks python3 pytorch

Last synced: 14 Nov 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: 11 Nov 2024

https://github.com/PengBoXiangShang/torchsketch

TPAMI 2022

cnn deep-learning gnn pytorch rnn sketch tcn

Last synced: 04 Nov 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