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/fcakyon/small-object-detection-benchmark

icip2022 paper: sahi benchmark on visdrone and xview datasets using fcos, vfnet and tood detectors

benchmark coco fcos icip2022 mmdetection object-detection pytorch sahi small-object-detection tood vfnet visdrone xview

Last synced: 30 Oct 2024

https://github.com/shinianzhihou/ChangeDetection

A framework for change detection using PyTorch

change-detection pytorch remote-sensing siamese unet

Last synced: 03 Aug 2024

https://github.com/GuoleiSun/CountSeg

Official code for "Object counting and instance segmentation with image-level supervision", in CVPR 2019 and TPAMI 2020

cvpr2019 instance-segmentation object-counting pytorch tpami2020 weakly-supervised-learning

Last synced: 05 Nov 2024

https://github.com/lukas-blecher/AutoMask

Automatic Rotoscoping/Masking tool powered with Machine Learning for Blender (WIP)

blender blender-addon blender-python deep-learning machine-learning mask-segment masking motion-tracking pytorch rotoscoping

Last synced: 03 Aug 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: 03 Aug 2024

https://github.com/benjs/nfnets_pytorch

Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".

deepmind nfnets pretrained-models pytorch sgd-adc weight-standardization

Last synced: 05 Aug 2024

https://github.com/mostafaelaraby/wavegan-pytorch

Pytorch Implementation of wavegan model to generate audio

ai-music gan music-generation pytorch wavegan wavegan-pytorch

Last synced: 04 Aug 2024

https://github.com/sidgan/etci-2021-competition-on-flood-detection

Experiments on Flood Segmentation on Sentinel-1 SAR Imagery with Cyclical Pseudo Labeling and Noisy Student Training

arxiv-papers deep-learning floods noisy-student pseudo-labeling python pytorch segmentation semi-supervised-learning

Last synced: 02 Nov 2024

https://github.com/chambliss/multilingual_ner

Applying BERT to named entity recognition in English and Russian.

bert english-language named-entity-recognition nlp-machine-learning pytorch russian-language spacy

Last synced: 14 Oct 2024

https://github.com/daddydrac/-deprecated-NVIDIA-GPU-Tensor-Core-Accelerator-PyTorch-OpenCV

Computer vision container that includes Jupyter notebooks with built-in code hinting, Anaconda, CUDA 11.8, TensorRT inference accelerator for Tensor cores, CuPy (GPU drop in replacement for Numpy), PyTorch, PyTorch geometric for Graph Neural Networks, TF2, Tensorboard, and OpenCV for accelerated workloads on NVIDIA Tensor cores and GPUs.

computer-vision cupy deep-learning image-processing machine-learning opencv pytorch tensorboard tensorflow2 tensorrt-inference-accelerator vision

Last synced: 06 Nov 2024

https://github.com/haruishi43/equilib

🌎→🗾Equirectangular (360/panoramic) image processing library for Python with minimal dependencies only using Numpy and PyTorch

360-degree camera cubemap-to-equirectangular equirectangular-images equirectangular-panorama equirectangular-projection numpy panorama perspective-projection python pytorch vr

Last synced: 26 Oct 2024

https://github.com/rusty1s/pyg_autoscale

Implementation of "GNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddings" in PyTorch

graph-neural-networks pytorch pytorch-geometric scalability

Last synced: 31 Oct 2024

https://github.com/liuzuxin/fsrl

🚀 A fast safe reinforcement learning library in PyTorch

cpo cvpo decision-making library ppo pytorch reinforcement-learning robotics sac safe-rl safety-critical trpo trustworthy-ai

Last synced: 30 Oct 2024

https://github.com/bfshi/AbSViT

Official code for "Top-Down Visual Attention from Analysis by Synthesis" (CVPR 2023 highlight)

attention classification cvpr pytorch segmentation vision-transformer

Last synced: 09 Nov 2024

https://github.com/DavexPro/pytorch-pose-estimation

PyTorch Implementation of Realtime Multi-Person Pose Estimation project.

computer-vision cvpr-2017 deep-learning human-pose-estimation python3 pytorch

Last synced: 07 Aug 2024

https://github.com/Yachay-AI/byt5-geotagging

Confidence and Byt5 - based geotagging model predicting coordinates from text alone.

coordinates deep-learning geo-location geotagging machine-learning neural-network nlp nlp-machine-learning python pytorch transformers

Last synced: 05 Nov 2024

https://github.com/chrisby/torchmtl

A lightweight module for Multi-Task Learning in pytorch.

deep-learning framework machine-learning mtl multi-task-learning python3 pytorch

Last synced: 31 Oct 2024

https://github.com/floydhub/dockerfiles

Deep Learning Dockerfiles

deep-learning docker pytorch tensorflow torch

Last synced: 06 Nov 2024

https://github.com/floydhub/floyd-cli

Command line tool for FloydHub - the fastest way to build, train, and deploy deep learning models

ai artificial-intelligence deep-learning floyd floyd-cli floydhub gpu machine-learning python pytorch tensorflow

Last synced: 31 Oct 2024

https://github.com/mkocabas/multi-person-tracker

Simple Multi Person Tracker implementation in PyTorch

multi-person-tracking object-tracking people-tracker pytorch sort

Last synced: 01 Nov 2024

https://github.com/lxuechen/private-transformers

A codebase that makes differentially private training of transformers easy.

deep-learning differential-privacy huggingface-transformers nlp pytorch transformers

Last synced: 27 Oct 2024

https://github.com/ndif-team/nnsight

The nnsight package enables interpreting and manipulating the internals of deep learned models.

interpretability machine-learning neural-networks python pytorch

Last synced: 15 Oct 2024

https://github.com/mouseland/facemap

Framework for predicting neural activity from mouse orofacial movements tracked using a pose estimation model. Package also includes singular value decomposition (SVD) of behavioral videos.

behavior deep-learning gui matlab-gui movie neuroscience pose-estimation pupil python pytorch rodents

Last synced: 06 Nov 2024

https://github.com/nvidia/torchfort

An Online Deep Learning Interface for HPC programs on NVIDIA GPUs

deep-learning fortran libtorch pytorch

Last synced: 30 Oct 2024

https://github.com/shchur/overlapping-community-detection

Implementation of "Overlapping Community Detection with Graph Neural Networks"

community-detection geometric-deep-learning graph-neural-networks overlapping-community-detection pytorch

Last synced: 01 Nov 2024

https://github.com/janfreyberg/pytorch-revgrad

A minimal pytorch package implementing a gradient reversal layer.

domain-adaptation gradient-reversal pytorch

Last synced: 26 Oct 2024

https://github.com/paul007pl/VRCNet

[CVPR 2021 Oral] Variational Relational Point Completion Network

mvp-dataset partial-point-clouds point-cloud-completion pytorch

Last synced: 28 Oct 2024

https://github.com/lxxue/frnn

Fixed Radius Nearest Neighbor Search on GPU

cuda nearest-neighbor-search pytorch

Last synced: 11 Nov 2024

https://github.com/jingxil/Neural-Decision-Forests

An implementation of the Deep Neural Decision Forests in PyTorch

pytorch

Last synced: 30 Oct 2024

https://github.com/cm-bf/featureflow

A state-of-the-art Video Frame Interpolation Method using feature flows blending. （CVPR 2020)

deep-learning feature-flow pytorch slow-motion video-frame-interpolation

Last synced: 07 Nov 2024

https://github.com/sayakpaul/caption-upsampling

This repository implements the idea of "caption upsampling" from DALL-E 3 with Zephyr-7B and gathers results with SDXL.

diffusers image-generation pytorch sdxl

Last synced: 31 Oct 2024

https://github.com/sony/sqvae

Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)

bayesian deep-generative-model generative-model gumbel-softmax machine-learning pytorch vae variational-autoencoder vector-quantization vq-vae

Last synced: 08 Nov 2024

https://github.com/aramis-lab/clinicadl

Framework for the reproducible processing of neuroimaging data with deep learning methods

alzheimer-disease brain-imaging convolutional-neural-network deep-learning medical-imaging neuroimaging python pytorch

Last synced: 12 Nov 2024

https://github.com/ozanciga/diffusion-for-beginners

denoising diffusion models, as simple as possible

dall-e diffusion imagen midjourney pytorch scheduler stable-diffusion

Last synced: 03 Aug 2024

https://github.com/CDOTAD/AlphaGAN-Matting

This project is an unofficial implementation of AlphaGAN: Generative adversarial networks for natural image matting published at the BMVC 2018

gan matting pytorch

Last synced: 03 Aug 2024

https://github.com/codeslake/Color_Transfer_Histogram_Analogy

[CGI 2020] Official PyTorch Implementation for "Deep Color Transfer using Histogram Analogy"

color-transfer deep-learning histogram-analogy pytorch

Last synced: 03 Aug 2024

https://github.com/chrisby/torchMTL

A lightweight module for Multi-Task Learning in pytorch.

deep-learning framework machine-learning mtl multi-task-learning python3 pytorch

Last synced: 03 Aug 2024

https://github.com/sanyam5/arc-pytorch

The first public PyTorch implementation of Attentive Recurrent Comparators

computer-vision deep-learning deep-neural-networks machine-learning pytorch recurrent-neural-networks

Last synced: 07 Aug 2024

https://github.com/jason-li-831202/Vehicle-CV-ADAS

The project can achieve FCWS, LDWS, and LKAS functions solely using only visual sensors. using YOLOv5 / YOLOv5-lite / YOLOv6 / YOLOv7 / YOLOv8 / YOLOv9 / EfficientDet and Ultra-Fast-Lane-Detection-v2 .

bytetrack efficientdet fcws lane-detection ldws lkas object-detection onnxruntime python pytorch tensorrt ultra-fast-lane-detection-v2 yolov10 yolov5 yolov5-lite yolov6 yolov7 yolov8 yolov9

Last synced: 09 Nov 2024

https://github.com/jla524/fromthetensor

From the Tensor to Stable Diffusion, a rough outline for a 9 week course.

deep-learning pytorch tinygrad

Last synced: 04 Aug 2024

https://github.com/ViCCo-Group/THINGSvision

Python package for extracting representations from state-of-the-art computer vision models

alignment cognitive-science computer-vision deep-learning neural-networks pytorch representations tensorflow

Last synced: 03 Aug 2024

https://github.com/szq0214/Un-Mix

Un-Mix: Rethinking Image Mixtures for Unsupervised Visual Representation Learning.

image-mixture pytorch self-supervised-learning smooth-decision-boundaries unsupervised-learning

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

https://github.com/drsleep/nas-segm-pytorch

Code for Fast Neural Architecture Search of Compact Semantic Segmentation Models via Auxiliary Cells, CVPR '19

camvid cityscapes cvpr-2019 dense-per-pixel-tasks depth-estimation neural-architecture-search pytorch real-time semantic-segmentation wacv2020

Last synced: 09 Nov 2024

https://github.com/devrimcavusoglu/pybboxes

Light weight toolkit for bounding boxes providing conversion between bounding box types and simple computations.

computer-vision deep-learning image-recognition machine-learning numpy python pytorch tensorflow

Last synced: 30 Oct 2024

https://github.com/clovaai/lffont

Official PyTorch implementation of LF-Font (Few-shot Font Generation with Localized Style Representations and Factorization) AAAI 2021

aaai2021 deep-learning font font-generation generative-models lf-font machine-learning pytorch

Last synced: 12 Nov 2024

https://github.com/liyuanlucasliu/ld-net

Efficient Contextualized Representation: Language Model Pruning for Sequence Labeling

contextualized-representation language-model model-compression named-entity-recognition ner pytorch sequence-labeling

Last synced: 09 Nov 2024

https://github.com/Graph-COM/GSAT

[ICML 2022] Graph Stochastic Attention (GSAT) for interpretable and generalizable graph learning.

deep-learning graph-neural-networks interpretability interpretable-machine-learning pytorch xai

Last synced: 08 Aug 2024

https://github.com/aidos-lab/pytorch-topological

A topological machine learning framework based on PyTorch

deep-learning pytorch topological-data-analysis topological-machine-learning

Last synced: 11 Oct 2024

https://github.com/xalanq/chinese-sentiment-classification

简单的中文文本情感分类 (MLP, CNN, RNN in PyTorch) - 2019 THU 人工智能导论作业

nlp pytorch

Last synced: 07 Nov 2024

https://github.com/coderonion/awesome-cuda-and-hpc

🔥🔥🔥 A collection of some awesome public CUDA, cuBLAS, TensorRT and High Performance Computing (HPC) projects.

awesome blas cublas cuda cudnn fortran gemm gpu hpc lapack llama llm mojo numpy openblas parallel-computing pytorch scipy tensorrt yolo

Last synced: 05 Oct 2024

https://github.com/guillaume-chevalier/Linear-Attention-Recurrent-Neural-Network

A recurrent attention module consisting of an LSTM cell which can query its own past cell states by the means of windowed multi-head attention. The formulas are derived from the BN-LSTM and the Transformer Network. The LARNN cell with attention can be easily used inside a loop on the cell state, just like any other RNN. (LARNN)

attention-is-all-you-need attention-mechanism attention-model lstm pytorch recurrent-neural-networks rnn

Last synced: 05 Nov 2024

https://github.com/wogong/pytorch-dann

A PyTorch implementation for Unsupervised Domain Adaptation by Backpropagation

deep-learning domain-adaptation generative-adversarial-network pytorch

Last synced: 27 Oct 2024

https://github.com/eambutu/snail-pytorch

Implementation of "A Simple Neural Attentive Meta-Learner" (SNAIL, https://arxiv.org/pdf/1707.03141.pdf) in PyTorch

cnn python pytorch snail

Last synced: 26 Oct 2024

https://github.com/zhenye-na/image-similarity-using-deep-ranking

🖼️ 𝖀𝖓𝖔𝖋𝖋𝖎𝖈𝖎𝖆𝖑 PyTorch implementation of "Learning Fine-grained Image Similarity with Deep Ranking" (arXiv:1404.4661)

deep-learning deep-ranking image-similarity python pytorch

Last synced: 12 Nov 2024

https://github.com/kashif/firedup

Clone of OpenAI's Spinning Up in PyTorch

deep-learning pytorch reinforcement-learning spinningup

Last synced: 12 Nov 2024

https://github.com/forestagostinelli/DeepCubeA

Code for DeepCubeA, a Deep Reinforcement Learning algorithm that can learn to solve the Rubik's cube.

astar-search deep-learning deep-reinforcement-learning deepcube deepcubea python3 pytorch reinforcement-learning rubiks-cube search

Last synced: 09 Aug 2024

https://github.com/foivospar/NED

PyTorch implementation for NED (CVPR 2022). It can be used to manipulate the facial emotions of actors in videos based on emotion labels or reference styles.

3dmm cvpr2022 deepfakes emotion expression-manipulation face face-edit gan image-generation pytorch talking-head

Last synced: 03 Aug 2024

https://github.com/erfanzar/EasyDeL

EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥

easydel flax gpt jax machine-learning mojo nlp optax pytorch transformers

Last synced: 03 Aug 2024

https://github.com/DeepRank/deeprank

This repository has been integrated in https://github.com/DeepRank/deeprank2

3d-cnn docking protein-protein-interaction pytorch

Last synced: 08 Aug 2024

https://github.com/gmkim-ai/Diffusion-Video-Autoencoders

An official implementation of "Diffusion Video Autoencoders: Toward Temporally Consistent Face Video Editing via Disentangled Video Encoding" (CVPR 2023) in PyTorch.

cvpr2023 deep-learning pytorch video

Last synced: 31 Oct 2024

https://github.com/xvjiarui/vfs

Rethinking Self-Supervised Correspondence Learning: A Video Frame-level Similarity Perspective, in ICCV 2021 (Oral)

correspondence pytorch representation-learning self-su video

Last synced: 27 Oct 2024

https://github.com/guillaume-chevalier/linear-attention-recurrent-neural-network

A recurrent attention module consisting of an LSTM cell which can query its own past cell states by the means of windowed multi-head attention. The formulas are derived from the BN-LSTM and the Transformer Network. The LARNN cell with attention can be easily used inside a loop on the cell state, just like any other RNN. (LARNN)

attention-is-all-you-need attention-mechanism attention-model lstm pytorch recurrent-neural-networks rnn

Last synced: 01 Nov 2024

https://github.com/stefanopini/simple-higherhrnet

Multi-person Human Pose Estimation with HigherHRNet in Pytorch, with TensorRT support

coco-dataset deep-learning higher-hrnet hrnet human-pose-estimation keypoint-detection mscoco-keypoint pytorch

Last synced: 07 Nov 2024

https://github.com/kakaobrain/trident

A performance library for machine learning applications.

ai deep-learning kakaobrain library machine-learning performance python pytorch trident triton

Last synced: 10 Nov 2024

https://github.com/hav4ik/Hydra

Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.

deep-learning multi-objective-optimization multi-task-learning neural-architecture-search pytorch

Last synced: 03 Aug 2024

https://github.com/nihalsid/ViewAL

[CVPR'20] Implementation for the paper "ViewAL: Active Learning with Viewpoint Entropy for Semantic Segmentation"

active-learning deep-learning pytorch semantic-segmentation

Last synced: 28 Oct 2024

https://github.com/ofa-sys/ofasys

OFASys: A Multi-Modal Multi-Task Learning System for Building Generalist Models

audio computer-vision deep-learning motion multimodal-learning multitask-learning nlp pretrained-models pytorch transformers vision-and-language

Last synced: 10 Oct 2024

https://github.com/vishal-v/100-days-of-ml-code

A 100 Day ML Challenge to learn and implement ML/DL concepts ranging from the basics of Machine Learning to advanced Deep Learning.

100-days-of-ml-code python pytorch sklearn tensorflow

Last synced: 10 Oct 2024

https://github.com/geyang/deep-auto-punctuation

a pytorch implementation of auto-punctuation learned character by character

auto-punctuation deep-learning engadget learning-punctuations pytorch reading-engadget-articles

Last synced: 11 Nov 2024

https://github.com/LeapLabTHU/Pseudo-Q

[CVPR 2022] Pseudo-Q: Generating Pseudo Language Queries for Visual Grounding

computer-vision cvpr2022 deep-learning multimodal-deep-learning pytorch vision-and-language visual-grounding

Last synced: 04 Nov 2024

https://github.com/Na-Z/attMPTI

[CVPR 2021] Few-shot 3D Point Cloud Semantic Segmentation

3d 3d-segmentation few-shot-learning point-cloud pytorch

Last synced: 28 Oct 2024

https://github.com/ianycxu/GCN-with-BERT

Graph Convolutional Networks (GCN) with BERT for Coreference Resolution Task [Pytorch][DGL]

bert bert-model coreference-resolution gcn gnn graph-convolutional-networks graph-neural-networks nlp pytorch

Last synced: 02 Nov 2024

https://github.com/TIGER-AI-Lab/ImagenHub

A one-stop library to standardize the inference and evaluation of all the conditional image generation models. (ICLR 2024)

deep-learning diffusion-models generative-ai generative-art image-editing image-generation pytorch stable-diffusion

Last synced: 06 Aug 2024

https://github.com/AllenXiangX/SnowflakeNet

(TPAMI 2023) Snowflake Point Deconvolution for Point Cloud Completion and Generation with Skip-Transformer

3dvision deep-learning iccv2021 point-cloud-completion pytorch tpami2023 transformers

Last synced: 03 Aug 2024

https://github.com/clovaai/mxfont

Official PyTorch implementation of MX-Font (Multiple Heads are Better than One: Few-shot Font Generation with Multiple Localized Experts) ICCV 2021

deep-learning font font-generation generative-models iccv2021 iccv21 machine-learning mx-font pytorch

Last synced: 12 Nov 2024

https://github.com/alvinwan/shiftresnet-cifar

ResNet with Shift, Depthwise, or Convolutional Operations for CIFAR-100, CIFAR-10 on PyTorch

cifar-10 cifar-100 classification compression convolutional-neural-networks pytorch resnet shift-operation

Last synced: 27 Oct 2024

https://github.com/jiwidi/Behavior-Sequence-Transformer-Pytorch

This is a pytorch implementation for the BST model from Alibaba https://arxiv.org/pdf/1905.06874.pdf

alibaba behavior behavior-sequence-transformer pytorch recommenders seq2seq seq2seq-attn transformers

Last synced: 05 Nov 2024

https://github.com/patrick-kidger/sympytorch

Turning SymPy expressions into PyTorch modules.

pytorch sympy

Last synced: 30 Oct 2024

https://github.com/RicherMans/GPV

Repository for our Interspeech2020 general-purpose voice activity detection (GPVAD) paper

machine-learning noise-robust-asr pytorch sound-activity speech-activity-detection voice-activity-detection

Last synced: 03 Aug 2024

https://github.com/prabhuomkar/iris

Open Source Photos Platform Powered by PyTorch

golang graphql machine-learning mongodb python pytorch rabbitmq react seaweedfs

Last synced: 02 Aug 2024

https://github.com/reiinakano/neural-painters-pytorch

PyTorch library for "Neural Painters: A learned differentiable constraint for generating brushstroke paintings"

deep-learning neural-painters pytorch style-transfer

Last synced: 01 Nov 2024

https://github.com/jiarenchang/realtimestereo

Attention-Aware Feature Aggregation for Real-time Stereo Matching on Edge Devices (ACCV, 2020)

psmnet pytorch real-time stereo-matching stereo-vision

Last synced: 11 Nov 2024

PyTorch Awesome Lists
Awesome-pytorch-list 704 awesome-semantic-segmentation-pytorch 35 Awesome-pytorch-list-CNVersion 692 awesome-colab-notebooks 525 CV-pretrained-model 103 awesome-AutoML-and-Lightweight-Models 150 openstl 43 Awesome-Denoise 164 ai4artists 286 arctic 34 awesome-cuda-and-hpc 427 awesome-yolov8-models 43 awesome-python-machine-learning-resources 1,884 awesome-adaptive-computation 173 awesome-NILM-with-code 47 awesome-dotnet-machine-learning 134 awesome-Multi-Document-Summarization 56 PyTorch-Guide 616 awesome-max-mojo 183 awesome-fastai 42 awesome-ai-tools 511 free-pytorch-resources 72
PyTorch Categories
Blogs 450 Frameworks 276 Pytorch & related libraries 231 Pytorch & related libraries｜Pytorch & 相关库 220 Research 198 Tutorials 192 文本数据和NLP 182 Uncategorized 161 数据可视化 131 机器学习框架 123 1.) Neural Architecture Search 122 Blogposts :sparkle: 119 图像数据与CV 106 Learning Resources 103 Learning 101 模型的可解释性 100 Tutorials, books, & examples 95 Interactive Learning & Tools 94 超参数优化和AutoML 93 Tutorials & books & examples｜教程 & 书籍 & 示例 93