PyTorch | Ecosyste.ms: Awesome

https://github.com/hkchengrex/scribble-to-mask

[CVPR 2021] MiVOS - Scribble to Mask module

computer-vision cvpr2021 deep-learning interactive-segmentation pytorch segmentation

Last synced: 10 Nov 2024

https://github.com/rasbt/pytorch-memory-optim

This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog post.

deep-learning llm memory-optimization pytorch vision vit

Last synced: 01 Nov 2024

https://github.com/unity-technologies/datasetinsights

Synthetic Dataset Insights

computer-vision machine-learning pytorch synthetic-datasets unity-simulation

Last synced: 07 Oct 2024

https://github.com/edisonleeeee/greatx

A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).

adversarial-attacks distribution-shift graph-convolutional-networks graph-neural-networks graph-reliability-toolbox inherent-noise pytorch pytorch-geometric

Last synced: 13 Nov 2024

https://github.com/dbaranchuk/memory-efficient-maml

Memory efficient MAML using gradient checkpointing

deep-learning gpu gradient-checkpointing maml meta-learning pytorch

Last synced: 07 Nov 2024

https://github.com/yuanzhoulvpi2017/quick_sentence_transformers

sentence-transformers to onnx 让sbert模型推理效率更快

bert pytorch sentence-transformers transformer transformers

Last synced: 08 Nov 2024

https://github.com/rockerboo/lora-inspector

LoRA (Low-Rank Adaptation) inspector for Stable Diffusion

analysis lora low-rank-approximation machine-learning pytorch

Last synced: 17 Nov 2024

https://github.com/project-monai/monai-deploy-app-sdk

MONAI Deploy App SDK offers a framework and associated tools to design, develop and verify AI-driven applications in the healthcare imaging domain.

ai deep-learning deploy dicom healthcare image-processing machine-learning medical-imaging ml ml-infrastructure ml-platform mlops model-deployment model-serving monai pipeline python pytorch workflow

Last synced: 10 Oct 2024

https://github.com/Atharva-Phatak/torchflare

TorchFlare is a simple, beginner-friendly, and easy-to-use PyTorch Framework train your models effortlessly.

deep-learning deep-neural-networks gans generative-adversarial-network image-classfication image-classification image-processing image-segmentation infrastructure kaggle machine-learning natural-language-processing python pytorch pytorch-model self-supervised-learning text-classification

Last synced: 15 Nov 2024

https://github.com/hellloxiaotian/deep-learning-on-image-denoising-an-overview

Deep Learning on Image Denoising: An overview (Neural Networks, 2020)

blind-denoising cnn denoising-network denoising-survery image-denoising keras keras-cnn low-level low-level-vision matconvnet python pytorch pytorch-cnn real-noise survery tensorflow tensorflow-cnn theano

Last synced: 11 Nov 2024

https://github.com/necla-ml/diff-jpeg

Official and maintained implementation of the paper "Differentiable JPEG: The Devil is in the Details" [WACV 2024].

differentiable-codec differentiable-jpeg image-codec image-compression jpeg pytorch ste surrogate wacv

Last synced: 10 Oct 2024

https://github.com/svishnu88/TGS-SaltNet

Kaggle | 21st place solution for TGS Salt Identification Challenge

computer-vision deep-learning fastai kaggle-competition pytorch

Last synced: 07 Aug 2024

https://github.com/Erlemar/wheat

Wheat Detection challenge on Kaggle

deep-learning hydra kaggle-competition object-detection pytorch pytorch-lightning torchvision

Last synced: 15 Nov 2024

https://github.com/huggingface/optimum-graphcore

Blazing fast training of 🤗 Transformers on Graphcore IPUs

fine-tuning graphcore machine-learning pytorch training transformers

Last synced: 02 Nov 2024

https://github.com/sevagh/demucs.cpp

C++17 port of Demucs v3 (hybrid) and v4 (hybrid transformer) models with ggml and Eigen3

blas demixing demucs eigen3 ggml guitar music-source-separation piano pytorch

Last synced: 27 Oct 2024

https://github.com/writer/fitbert

Use BERT to Fill in the Blanks

bert fill-in-the-blank nlp python pytorch

Last synced: 29 Sep 2024

https://github.com/kefirski/pytorch_TDNN

Time Delayed NN implemented in pytorch

cnn deep-learning python pytorch tdnn

Last synced: 14 Nov 2024

https://github.com/dengfenglai321/YOLOv4-Hat-detection

基于YOLOv4的安全帽佩戴检测

hat-detection pytorch yolov4

Last synced: 09 Nov 2024

https://github.com/theos-ai/easy-yolov7

This a clean and easy-to-use implementation of YOLOv7 in PyTorch, made with ❤️ by Theos AI.

computer-vision deep-learning machine-learning neural-networks object-detection python pytorch yolov5 yolov7

Last synced: 09 Nov 2024

https://github.com/rose-stl-lab/dyffusion

[NeurIPS 2023] A Dynamics-informed Diffusion Model for Spatiotemporal Forecasting

deep-learning diffusion diffusion-models ensemble-forecasts machine-learning neurips neurips-2023 probabilistic-forecasting pytorch pytorch-lightning spatiotemporal-forecasting

Last synced: 09 Nov 2024

https://github.com/yusugomori/deeplearning-keras-tf2-torch

詳解ディープラーニング第2版

deep-learning keras pytorch tensorflow2

Last synced: 07 Nov 2024

https://github.com/spcl/daceml

A Data-Centric Compiler for Machine Learning

compiler cuda deep-learning fpga high-performance-computing machine-learning pytorch

Last synced: 06 Nov 2024

https://github.com/kevinzakka/form2fit

[ICRA 2020] Train generalizable policies for kit assembly with self-supervised dense correspondence learning.

deep-learning perception pytorch robotics self-supervised-learning

Last synced: 28 Oct 2024

https://github.com/lyeoni/pretraining-for-language-understanding

Pre-training of Language Models for Language Understanding

language-model language-modeling language-understanding nlp pytorch pytorch-tutorial

Last synced: 06 Nov 2024

https://github.com/intel/torch-ccl

oneCCL Bindings for Pytorch*

oneccl pytorch

Last synced: 30 Oct 2024

https://github.com/d4l3k/torchdrive

torch models and primitives for birdseye view and voxel models for self driving tasks.

pytorch self-driving

Last synced: 14 Nov 2024

https://github.com/acl21/deep-active-learning-pytorch

A PyTorch toolkit with 8 popular deep active learning query methods implemented.

deep-active-learning image-classification pytorch toolkit

Last synced: 05 Nov 2024

https://github.com/anshulpaigwar/Attentional-PointNet

Attentional-PointNet is Deep-Neural-Network architecture for 3D object detection in point clouds

3d-object-detection attention-mechanism autonomous-vehicles object-detection point-cloud pointnet pytorch pytorch-implmention

Last synced: 28 Oct 2024

https://github.com/rose-stl-lab/torchts

Time series forecasting with PyTorch

deep-learning machine-learning pytorch pytorch-lightning time-series

Last synced: 15 Nov 2024

https://github.com/erogol/fftnet

FFTNet vocoder implementation

deep-learning fftnet pytorch text2speech vocoder

Last synced: 22 Oct 2024

https://github.com/lromul/argus

Lightweight library for training neural networks in PyTorch

deep-learning machine-learning neural-network python pytorch

Last synced: 10 Oct 2024

https://github.com/huawei-noah/multi_hyp_cc

[CVPR2020] A Multi-Hypothesis Approach to Color Constancy

color-constancy convolutional-neural-networks cvpr cvpr2020 pytorch whitebalance

Last synced: 13 Nov 2024

https://github.com/shchur/ifl-tpp

Implementation of "Intensity-Free Learning of Temporal Point Processes" (Spotlight @ ICLR 2020)

normalizing-flows pytorch temporal-point-processes

Last synced: 09 Nov 2024

https://github.com/dwromero/wavelet_networks

Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs/2006.05259

cnns equivariance group-convolution pytorch time-series

Last synced: 13 Nov 2024

https://github.com/asteroid-team/asteroid-filterbanks

Asteroid's filterbanks :rocket:

asteroid-filterbanks audio audio-processing deep-learning filterbanks python3 pytorch

Last synced: 05 Nov 2024

https://github.com/yaox12/byol-pytorch

PyTorch implementation of "Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning" with DDP and Apex AMP

byol byol-pytorch pretrained-model pytorch pytorch-implementation self-supervised-learning

Last synced: 28 Oct 2024

https://github.com/dsaurus/tensor4d

This is the official implementation of Tensor4D: Efficient Neural 4D Decomposition for High-fidelity Dynamic Reconstruction and Rendering.

3d-reconstruction 4d machine-learning nerf pytorch rendering

Last synced: 09 Nov 2024

https://github.com/kefirski/pytorch_Highway

Highway network implemented in pytorch

highway-network python pytorch

Last synced: 14 Nov 2024

https://github.com/ndrplz/differentiable-renderer

Rastering algorithm to approximate the rendering of a 3D model silhouette in a fully differentiable way.

differentiable-rendering eccv-2018 python pytorch rendering rendering-engine tensorflow tensorflow-models

Last synced: 12 Nov 2024

https://github.com/lucasjinreal/ssds_pytorch

Multiple basenet MobileNet v1,v2, ResNet combined with SSD detection method and it's variants such as RFB, FSSD etc.

mobilenet mobilenetv2 pytorch ssd

Last synced: 06 Nov 2024

https://github.com/imdeepmind/neuralpy

NeuralPy: A Keras like deep learning library works on top of PyTorch

data-science deep-learning keras library machine-learning neural-network neuralpy neuralpy-torch python pytorch

Last synced: 13 Nov 2024

https://github.com/vitae-transformer/vitae-transformer-scene-text-detection

A comprehensive list [GoMatching@NeurIPS'24, DeepSolo(++)@ CVPR'23, DPText-DETR@AAAI'23, I3CL@IJCV'22] of our research works related to scene text detection, spotting, etc., including papers, codes.

deep-learning ocr pytorch scene-text-detection scene-text-spotting vision-transformer

Last synced: 14 Nov 2024

https://github.com/adamdad/knowledgefactor

[ECCV2022] Factorizing Knowledge in Neural Networks

deep-learning distillation eccv2022 kd knowldge-distillation knowledge-transfer multitask-learning pytorch

Last synced: 16 Nov 2024

https://github.com/mukosame/aoda

Official implementation of "Adversarial Open Domain Adaptation for Sketch-to-Photo Synthesis"(WACV 2022/CVPRW 2021)

computer-vision cvpr cvpr2021 deep-learning gan gans image-generation image-manipulation image2image open-domain pytorch sketch wacv wacv2022

Last synced: 07 Nov 2024

https://github.com/QY-H00/attention-interpolation-diffusion

[NeurIPS 2024] Official Implementation of Attention Interpolation of Text-to-Image Diffusion

deep-learning diffusion image-interpolation pytorch stable-diffusion text-to-image-generation

Last synced: 31 Oct 2024

https://github.com/guyyariv/TempoTokens

This repo contains the official PyTorch implementation of: Diverse and Aligned Audio-to-Video Generation via Text-to-Video Model Adaptation

ai-art audio-to-video audio-visual deep-learning diffusion-models generative-ai modelscope pytorch video-synthesis

Last synced: 05 Aug 2024

https://github.com/lartpang/hdfnet

(ECCV 2020) Hierarchical Dynamic Filtering Network for RGB-D Salient Object Detection

eccv eccv2020 image-processing imageprocessing imagesegmentation pytorch saliency saliency-detection salient-object-detection segmentation

Last synced: 19 Nov 2024

https://github.com/val-iisc/sketch-parse

Code, demos and data for SketchParse (a neural network for sketch segmentation). Paper:

deep-learning pytorch semantic-segmentation sketches

Last synced: 12 Aug 2024

https://github.com/ajhamdi/mvtorch

a Pytorch library for multi-view 3D understanding and generation

3d 3d-deep-learning deep learning multi-view multi-view-geometry multi-view-learning nerf pytorch pytorch3d

Last synced: 28 Oct 2024

https://github.com/subeeshvasu/2018_subeesh_epsr_eccvw

Project page of the paper 'Analyzing Perception-Distortion Tradeoff using Enhanced Perceptual Super-resolution Network' (ECCVW 2018)

deep-learning epsr pirm-sr pytorch super-resolution

Last synced: 06 Nov 2024

https://github.com/sahagobinda/GPM

Official [ICLR] Code Repository for "Gradient Projection Memory for Continual Learning"

computer-vision continual-learning deep-learning optimizaiton pytorch

Last synced: 15 Nov 2024

https://github.com/wovchena/text-detection-fots.pytorch

FOTS text detection branch reimplementation, hmean: 83.3%

cvpr2018 fots pytorch text-detection

Last synced: 22 Oct 2024

https://github.com/ovshake/stable-fashion

Stable Fashion: A prompt based virtual try on repository

fashion pytorch stable-diffusion try-on

Last synced: 17 Nov 2024

https://github.com/hellloxiaotian/acnet

Asymmetric CNN for image super-resolution (IEEE Transactions on Systmes, Man, and Cybernetics: Systems 2021)

blind-super-resolution cnn cnn-pytorch deep-learning deep-network deep-neural-networks image-processing low-level-vision python2 python27 pytorch pytorch-cnn super-resolution

Last synced: 12 Oct 2024

https://github.com/giotto-ai/giotto-deep

Deep learning made topological.

computational-geometry deep-learning image-processing nlp pytorch tda

Last synced: 18 Nov 2024

https://github.com/tommyip/mamba2-minimal

Minimal Mamba-2 implementation in PyTorch

llm machine-learning mamba pytorch ssm

Last synced: 17 Nov 2024

https://github.com/ternaus/angiodysplasia-segmentation

Wining solution and its further development for MICCAI 2017 Endoscopic Vision Challenge Angiodysplasia Detection and Localization

computer-vision deep-learning image-segmentation medical-imaging python pytorch

Last synced: 28 Oct 2024

https://github.com/eva-n27/BERT-for-Chinese-Question-Answering

bert pytorch question-answering

Last synced: 02 Nov 2024

https://github.com/dbraun/pytorchtop

GPU PyTorch TOP in TouchDesigner with CUDA-enabled OpenCV

cuda libtorch opencv pytorch touchdesigner

Last synced: 08 Nov 2024

https://github.com/li012589/NeuralRG

Pytorch source code for arXiv paper Neural Network Renormalization Group, a generative model using variational renormalization group and normalizing flow.

ising-model normalizing-flow pytorch renormalization-group

Last synced: 05 Aug 2024

https://github.com/pietrolesci/energizer

An active learning library for Pytorch based on Lightning-Fabric.

active-learning deep-learning python pytorch pytorch-lightning

Last synced: 19 Nov 2024

https://github.com/nvidia/nvimagecodec

A nvImageCodec library of GPU- and CPU- accelerated codecs featuring a unified interface

computer-vision cpp cuda dali data-processing deep-learning fast-data-pipeline gpu image-processing machine-learning nvidia python pytorch

Last synced: 13 Nov 2024

https://github.com/chongruo/pytorch-HED

An implementation of HED in pytorch

pytorch

Last synced: 15 Nov 2024

https://github.com/ahmdtaha/knowledge_evolution

(CVPR-Oral 2021) PyTorch implementation of Knowledge Evolution approach and Split-Nets

classification computer-vision deep-learning deep-neural-networks few-shot-learning knowledge-distillation knowledge-evolution machine-learning python3 pytorch slim-networks

Last synced: 06 Nov 2024

https://github.com/2014gaokao/pedestrian-attribute-recognition-with-GCN

Pytorch implementation of pedestrian attribute recognition with graph convolutional network

attribute cnn gcn pytorch

Last synced: 02 Nov 2024

https://github.com/cogitare-ai/cogitare

🔥 Cogitare - A Modern, Fast, and Modular Deep Learning and Machine Learning framework for Python

deep-learning framework machine-learning modular-deep-learning pytorch toolbox toolchain

Last synced: 14 Nov 2024

https://github.com/borchero/natural-posterior-network

Official Implementation of "Natural Posterior Network: Deep Bayesian Predictive Uncertainty for Exponential Family Distributions" (ICLR, 2022)

bayesian-deep-learning deep-learning machine-learning pytorch uncertainty-estimation

Last synced: 27 Oct 2024

https://github.com/matifali/dockerdl

Deep Learning Docker Image

cuda deep-learning docker jupyter numpy python pytorch tensorflow

Last synced: 14 Nov 2024

https://github.com/vocalpy/vak

A neural network framework for researchers studying acoustic communication

animal-communication animal-vocalizations bioacoustic-analysis bioacoustics birdsong python python3 pytorch spectrograms speech-processing torch torchvision vocalizations

Last synced: 13 Nov 2024

https://github.com/alishdipani/neural-style-transfer-audio

This is PyTorch Implementation of Neural Style Transfer Algorithm which is modified for Audios.

audio-processing convolutional-neural-networks fourier-transform neural-style-transfer pytorch pytorch-implmention

Last synced: 15 Oct 2024

https://github.com/kabouzeid/point2vec

Self-Supervised Representation Learning on Point Clouds (GCPR 2023 | T4V Workshop @ CVPR 2023)

lightning machine-learning point-cloud pytorch self-supervised-learning transformer vision vision-transformer

Last synced: 27 Oct 2024

https://github.com/salu133445/ismir2019tutorial

Website for tutorial "Generating Music with GANs: An Overview and Case Studies"

gan generative-adversarial-network machine-learning music music-generation pytorch

Last synced: 29 Oct 2024

https://github.com/holywu/vs-realesrgan

Real-ESRGAN function for VapourSynth

pytorch real-esrgan tensorrt vapoursynth

Last synced: 01 Oct 2024

https://github.com/alishdipani/Neural-Style-Transfer-Audio

This is PyTorch Implementation of Neural Style Transfer Algorithm which is modified for Audios.

audio-processing convolutional-neural-networks fourier-transform neural-style-transfer pytorch pytorch-implmention

Last synced: 07 Aug 2024

https://github.com/m-tassano/dvdnet

DVDnet: A Simple and Fast Network for Deep Video Denoising

cnn cnn-architecture deep-learning denoising python3 pytorch video

Last synced: 03 Nov 2024

https://github.com/aimagelab/art2real

Art2Real: Unfolding the Reality of Artworks via Semantically-Aware Image-to-Image Translation. CVPR 2019

art2real computer-vision cvpr2019 deep-learning image-to-image-translation pytorch

Last synced: 07 Nov 2024

https://github.com/ActiveVisionLab/gaussctrl

[ECCV 2024] GaussCtrl: Multi-View Consistent Text-Driven 3D Gaussian Splatting Editing

3d 3d-editing computer-vision diffusion-models eccv2024 gaussian-splatting pytorch

Last synced: 14 Nov 2024

https://github.com/MaverickPeter/DiSCO-pytorch

Differentiable Scan Context with Orientation

icra2021 place-recognition pytorch robotics

Last synced: 28 Oct 2024

https://github.com/inferno-pytorch/speedrun

Research code need not be ugly.

deep-learning neural-networks pytorch research

Last synced: 05 Nov 2024

https://github.com/yuchenlin/lstm_sentence_classifier

LSTM-based Models for Sentence Classification in PyTorch

lstm-model nlp pytorch sentence-classification

Last synced: 28 Oct 2024

https://github.com/HolyWu/vs-realesrgan

Real-ESRGAN function for VapourSynth

pytorch real-esrgan tensorrt vapoursynth

Last synced: 03 Nov 2024

https://github.com/Uehwan/3-D-Scene-Graph

3D scene graph generator implemented in Pytorch.

3d-models 3d-scene-graph computer-vision deep-learning deeplearning environmental-modelling intelligence pytorch robot robotics scene-graph scenegraph

Last synced: 02 Nov 2024

https://github.com/fabioperez/pytorch-examples

Tutorials, examples, and projects implemented with PyTorch

deep-learning deep-neural-networks examples machine-learning pytorch tutorials

Last synced: 05 Nov 2024

https://github.com/sar-gupta/convisualize_nb

Visualisations for Convolutional Neural Networks in Pytorch

cnn convolutional-neural-networks deep-learning neural-network neural-networks python pytorch visualization

Last synced: 07 Nov 2024

https://github.com/lromul/argus-tgs-salt

Kaggle | 14th place solution for TGS Salt Identification Challenge

deep-learning kaggle kaggle-competition neural-network python pytorch segmentation

Last synced: 22 Oct 2024

https://github.com/tingiskhan/pyfilter

Particle filtering and sequential parameter inference in Python

bayesian-inference particle-filter pytorch sequential-monte-carlo time-series

Last synced: 13 Nov 2024

https://github.com/j-min/mocha-pytorch

PyTorch Implementation of "Monotonic Chunkwise Attention" (ICLR 2018)

attention-mechanism hard-attention monotonic-attention pytorch seq2seq

Last synced: 19 Nov 2024

https://github.com/isaaccorley/pytorch-enhance

Open-source Library of Image Super-Resolution Models, Datasets, and Metrics for Benchmarking or Pretrained Use

artificial-intelligence computer-vision deep-learning pytorch super-resolution

Last synced: 22 Oct 2024

https://github.com/imirzadeh/stable-continual-learning

Towards increasing stability of neural networks for continual learning: https://arxiv.org/abs/2006.06958.pdf (NeurIPS'20)

catastrophic-forgetting continual-learning deep-learning lifelong-learning pytorch

Last synced: 15 Nov 2024

https://github.com/xxxnell/spatial-smoothing

(ICML 2022) Official PyTorch implementation of “Blurs Behave Like Ensembles: Spatial Smoothings to Improve Accuracy, Uncertainty, and Robustness”.

bayesian-deep-learning bayesian-neural-networks neural-network pytorch robustness uncertainty uncertainty-estimation uncertainty-quantification

Last synced: 15 Nov 2024

https://github.com/aksub99/moldqn-pytorch

A PyTorch Implementation of "Optimization of Molecules via Deep Reinforcement Learning".

chemistry deep-learning dqn-pytorch drug-discovery inverse-design machine-learning materials-informatics materials-science molecule python pytorch pytorch-implementation pytorch-rl reinforcement-learning

Last synced: 09 Oct 2024

https://github.com/howardyclo/pytorch-seq2seq-example

Fully batched seq2seq example based on practical-pytorch, and more extra features.

attention-seq2seq checkpoint fixed-embedding glove-embeddings jupyter-notebook nlp nlp-machine-learning pretrained-embedding pytorch pytorch-nlp-tutorial pytorch-tutorial seq2seq shared-embedding tensorboard tensorboard-visualization tie-embedding

Last synced: 09 Nov 2024

https://github.com/gsurma/jetson

Self-driving AI toy car 🤖🚗.

autonomous-driving autopilot cnn diy end-to-end jetbot jetracer jetson jetson-nano nano nvidia nvidia-jetson-nano python pytorch self-driving-car self-driving-cars vision waveshare

Last synced: 18 Nov 2024

https://github.com/LukasHedegaard/pytorch-benchmark

Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption

benchmark deep-learning flops gpu jetson python pytorch timing-analysis

Last synced: 09 Aug 2024

https://github.com/wandering007/nasnet-pytorch

A neat pytorch implementation of nasnet and the ported weights from tensorflow

nasnet pytorch

Last synced: 31 Oct 2024

https://github.com/pyrates-neuroscience/pyrates

Open-source, graph-based Python code generator and analysis toolbox for dynamical systems (pre-implemented and custom models). Most pre-implemented models belong to the family of neural population models.

code-generation delayed-differential-equation differential-equations dynamical-systems fortran90 julia matlab network-simulator neural-networks numpy parameter-search python pytorch scientific-computing scientific-research simulations tensorflow

Last synced: 11 Oct 2024

https://github.com/pyrates-neuroscience/PyRates

Open-source, graph-based Python code generator and analysis toolbox for dynamical systems (pre-implemented and custom models). Most pre-implemented models belong to the family of neural population models.

code-generation delayed-differential-equation differential-equations dynamical-systems fortran90 julia matlab network-simulator neural-networks numpy parameter-search python pytorch scientific-computing scientific-research simulations tensorflow