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/gatheluck/FourierHeatmap

Unofficial pytorch implementation of Fourier Heat Map proposed in 'A Fourier Perspective on Model Robustness in Computer Vision' [Yin+, NeurIPS2019]

fourier-analysis heatmap pytorch robustness

Last synced: 03 Aug 2024

https://github.com/hunto/image_classification_sota

Training ImageNet / CIFAR models with sota strategies and fancy techniques such as ViT, KD, Rep, etc.

cifar image-classification imagenet kd nas pruning pytorch rep transformer vit

Last synced: 03 Aug 2024

https://github.com/kacky24/stylenet

A pytorch implemention of "StyleNet: Generating Attractive Visual Captions with Styles"

caption cnn dlhacks image-captioning lstm microsoft pytorch pytorch-implemention stylenet

Last synced: 01 Aug 2024

https://github.com/EMUNES/Auto-Subtitle-File-Generation

Generate subtitle files with timelines in an automatic way.

automatic deep-learning generator pytorch subtitles

Last synced: 04 Aug 2024

https://github.com/granularai/fabric

Urban change model designed to identify changes across 2 timestamps

change-detection cnn deep-learning pytorch satellite-imagery sentinel-2 unet unet-3d

Last synced: 01 Aug 2024

https://github.com/cthoyt/class-resolver

🔍 Lookup classes and instantiate them with style

optuna python pytorch utility

Last synced: 03 Aug 2024

https://github.com/kaituoxu/X-Punctuator

A PyTorch implementation of a punctuation prediction system using (B)LSTM, which automatically adds suitable punctuation into text without punctuation.

lstm punctuation pytorch

Last synced: 03 Aug 2024

https://github.com/tonyduan/diffusion

From-scratch diffusion model implemented in PyTorch.

diffusion generative-models pytorch

Last synced: 31 Jul 2024

https://github.com/ducha-aiki/google-retrieval-challenge-2019-fastai-starter

fast.ai starter kit for Google Landmark Retrieval 2019 challenge

cnn fastai kaggle pytorch retrieval

Last synced: 02 Aug 2024

https://github.com/lukasliebel/dfc2020_baseline

Simple Baseline for the IEEE GRSS Data Fusion Contest 2020

deep-learning earth-observation pytorch remote-sensing

Last synced: 07 Aug 2024

https://github.com/ap229997/Conditional-Batch-Norm

Pytorch implementation of NIPS 2017 paper "Modulating early visual processing by language"

cbn modulated-resnet pytorch vqa

Last synced: 03 Aug 2024

https://github.com/Confusezius/ICCV2019_MIC

(ICCV 2019) This repo contains code for "MIC: Mining Interclass Characteristics for Improved Metric Learning", which proposes an auxiliary training task to explain away intra-class variations.

cars196 computer-vi cub200 deep-lear deep-metric-learning metric- pku-vehicle pytorch shop-clothes-retrieval

Last synced: 03 Aug 2024

https://github.com/moskomule/anatome

Ἀνατομή is a PyTorch library to analyze representation of neural networks

neural-networks pytorch pytorch-library representation-learning

Last synced: 03 Aug 2024

https://github.com/LiyuanLucasLiu/Torch-Scope

A Toolkit for Training, Tracking, Saving Models and Syncing Results

gpu-memory logger pytorch tensorboard toolkit

Last synced: 03 Aug 2024

https://github.com/MrtnMndt/OpenVAE_ContinualLearning

Open-source code for our paper: Unified Probabilistic Deep Continual Learning through Generative Replay and Open Set Recognition

continual-learning deep-learning open-set-recognition pytorch variational-autoencoder

Last synced: 03 Aug 2024

https://github.com/fregu856/ebms_3dod

Official implementation of "Accurate 3D Object Detection using Energy-Based Models", CVPR Workshops 2021.

3d-object-detection computer-vision deep-learning energy-based-model machine-learning object-detection pytorch

Last synced: 31 Jul 2024

https://github.com/MzeroMiko/mamba-mini

An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivation. It is probably the code which is the most close to selective_scan_cuda in mamba.

efficient mamba pytorch selective-scan

Last synced: 31 Jul 2024

https://github.com/leftthomas/DGCNN

A PyTorch implementation of DGCNN based on AAAI 2018 paper "An End-to-End Deep Learning Architecture for Graph Classification"

gcn gnn graph graph-classification graph-convolutional-networks pytorch

Last synced: 31 Jul 2024

https://github.com/abdoelsayed2016/TNCR_Dataset

Deep learning, Convolutional neural networks, Image processing, Document processing, Table detection, Page object detection, Table classification. https://www.sciencedirect.com/science/article/pii/S0925231221018142

classification cnn convolutional-neural-networks deep-learning image-classification mmdetection object-detection object-recognition pytorch table table-detection table-recognition

Last synced: 03 Aug 2024

https://github.com/swyoon/pytorch-minimal-gaussian-process

A minimal implementation of Gaussian process regression in PyTorch

gaussian-process gaussian-process-regression gaussian-processes pytorch

Last synced: 02 Aug 2024

https://sirui-xu.github.io/STARS/

[ECCV 2022 oral] Official PyTorch implementation of the paper "Diverse Human Motion Prediction Guided by Multi-Level Spatial-Temporal Anchors"

deep-learning disentangled-representations eccv2022 generative-model graph-neural-networks human-motion-prediction motion-forecasting motion-prediction pytorch

Last synced: 02 Aug 2024

https://github.com/Sirui-Xu/STARS

[ECCV 2022 oral] Official PyTorch implementation of the paper "Diverse Human Motion Prediction Guided by Multi-Level Spatial-Temporal Anchors"

deep-learning disentangled-representations eccv2022 generative-model graph-neural-networks human-motion-prediction motion-forecasting motion-prediction pytorch

Last synced: 02 Aug 2024

https://github.com/Lightning-Universe/lightning-ColossalAI

Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI

lightning python pytorch

Last synced: 02 Aug 2024

https://github.com/jw9730/hot

[NeurIPS'21] Higher-order Transformers for sets, graphs, and hypergraphs, in PyTorch

equivariance gnn graph hypergraph pytorch self-attention transformer

Last synced: 31 Jul 2024

https://github.com/val-iisc/SDAT

[ICML 2022]Source code for "A Closer Look at Smoothness in Domain Adversarial Training",

adversarial-training dann domain-adaptation icml-2022 pytorch sharpness-aware-minimization

Last synced: 03 Aug 2024

https://github.com/baiksung/MeTAL

Official PyTorch implementation of "Meta-Learning with Task-Adaptive Loss Function for Few-Shot Learning" (ICCV2021 Oral)

computer-vision deep-learning few-shot-learning iccv machine-learning maml meta-learning pytorch

Last synced: 03 Aug 2024

https://github.com/speediedan/finetuning-scheduler

A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.

artificial-intelligence fine-tuning finetuning machine-learning neural-networks pytorch pytorch-lightning superglue transfer-learning

Last synced: 01 Aug 2024

https://github.com/KushajveerSingh/resize_network_cv

PyTorch implementation of the paper "Learning to Resize Images for Computer Vision Tasks" on Imagenette and Imagewoof datasets

computer-vision pytorch pytorch-implementation pytorch-lightning

Last synced: 03 Aug 2024

https://github.com/youngfly11/LCMCG-PyTorch

AAAI2020-The official implementation of "Learning Cross-modal Context Graph for Visual Grounding"

aaai2020 pytorch

Last synced: 04 Aug 2024

https://github.com/Lee-Gihun/MixCo-Mixup-Contrast

(NeurIPS 2020 Workshop on SSL) Official Implementation of "MixCo: Mix-up Contrastive Learning for Visual Representation"

deep-learning mix-up pytorch self-supervised-learning

Last synced: 02 Aug 2024

https://github.com/mboudiaf/pytorch-meta-dataset

A non-official 100% PyTorch implementation of META-DATASET benchmark for few-shot classification

few-shot-classifcation few-shot-learning meta-dataset pytorch tfrecorddataset

Last synced: 03 Aug 2024

https://github.com/lab-cosmo/sphericart

Multi-language library for the calculation of spherical harmonics in Cartesian coordinates

geometric-deep-learning python pytorch spherical-harmonics

Last synced: 02 Aug 2024

https://github.com/saturncloud/dask-pytorch-ddp

dask-pytorch-ddp is a Python package that makes it easy to train PyTorch models on dask clusters using distributed data parallel.

computer-vision dask deep-learning distributed-computing machine-learning nlp pytorch

Last synced: 03 Aug 2024

https://github.com/jinglescode/python-signal-processing

splearn: package for signal processing and machine learning with Python. Contains tutorials on understanding and applying signal processing.

machine-learning numpy python python-tutorial pytorch pytorch-tutorial scipy sequence signal-processing

Last synced: 31 Jul 2024

https://github.com/unixwzrd/oobabooga-macOS

Optimizing performance, building and installing packages required for oobabooga, AI and Data Science on Apple Silicon GPU. The goal is to optimize wherever possible, from the ground up.

blas journal llama-cpp-python macos numpy oobabooga pytorch

Last synced: 09 Aug 2024

https://github.com/aangelopoulos/conformal-risk

Conformal prediction for controlling monotonic risk functions. Simple accompanying PyTorch code for conformal risk control in computer vision and natural language processing.

computer-vision conformal conformal-prediction natural-language-processing python pytorch pytorch-implementation uncertainty-estimation uncertainty-quantification

Last synced: 01 Aug 2024

https://github.com/vexilligera/strokenet

a neural network that draws digits using strokes

deep-learning nodejs pytorch reinforcement-learning

Last synced: 02 Aug 2024

https://github.com/ztxtech/Time-Evidence-Fusion-Network

Official implementation of "Time Evidence Fusion Network: Multi-source View in Long-Term Time Series Forecasting" (https://arxiv.org/abs/2405.06419)

cuda deep-learning machine-learning macos neural-network neural-networks pytorch time-series time-series-analysis time-series-forecasting time-series-prediction uestc

Last synced: 01 Aug 2024

https://github.com/xierc/Semi_Human_Pose

[ICCV-2021] An Empirical Study of the Collapsing Problem in Semi-Supervised 2D Human Pose Estimation

consistency-regularization human-pose-estimation pytorch semi-supervised-learning

Last synced: 08 Aug 2024

https://github.com/customdiffusion360/custom-diffusion360

CustomDiffusion360: Customizing Text-to-Image Diffusion with Camera Viewpoint Control

camera-pose-control computer-vision customization generative-model pytorch stable-diffusion text-to-image

Last synced: 31 Jul 2024

https://github.com/byminji/SLTtrack

Official Implementation of Towards Sequence-Level Training for Visual Tracking (ECCV 2022)

eccv2022 pytorch reinforcement-learning single-object-tracking sot tracking visual-tracking vot

Last synced: 01 Aug 2024

https://github.com/Toytiny/RaFlow

Self-Supervised Scene Flow Estimation with 4-D Automotive Radar (RA-L & IROS'22)

automotive-radar autonomous-driving deep-learning point-cloud pytorch scene-flow-estimation self-supervised

Last synced: 31 Jul 2024

https://github.com/OIdiotLin/torchtracer

A python package for visualization and storage management in a pytorch AI task.

lab machine-learning pytorch tools

Last synced: 03 Aug 2024

https://github.com/angeluriot/Generative_adversarial_network

A deep learning model that can create high quality images by training on a dataset.

deep-learning gan generative-ai pytorch

Last synced: 01 Aug 2024

https://github.com/dsgiitr/adversarial_lab

Web-based Tool for visualisation and generation of adversarial examples by attacking ImageNet Models like VGG, AlexNet, ResNet etc.

adversarial-attacks computer-vision flask html-css-javascript imagenet machine-learning python pytorch visualization

Last synced: 02 Aug 2024

https://github.com/csiro-robotics/TCE

This repository contains the code implementation used in the paper Temporally Coherent Embeddings for Self-Supervised Video Representation Learning (TCE).

action-recognition computer-vision contrastive-learning contrastive-loss deep-learning embeddings hmdb51 kinetics-datasets metric-learning pytorch representation-learning self-supervised-learning tsne-visualisations ucf-101

Last synced: 02 Aug 2024

https://github.com/Harry24k/CW-pytorch

A pytorch implementation of "Towards Evaluating the Robustness of Neural Networks"

adversarial-attacks deep-learning pytorch

Last synced: 01 Aug 2024

https://github.com/francois-rozet/torchist

NumPy-style histograms in PyTorch

histogram numpy pytorch

Last synced: 03 Aug 2024

https://github.com/oke-aditya/quickvision

An Easy To Use PyTorch Computer Vision Library

computer-vision deep-learning pytorch pytorch-lightning

Last synced: 03 Aug 2024

https://github.com/MrinmoiHossain/Udacity-Deep-Learning-Nanodegree

The course is contained knowledge that are useful to work on deep learning as an engineer. Simple neural networks & training, CNN, Autoencoders and feature extraction, Transfer learning, RNN, LSTM, NLP, Data augmentation, GANs, Hyperparameter tuning, Model deployment and serving are included in the course.

convolutional-networks convolutional-neural-networks deep-learning gans generative-adversarial-network long-short-term-memory-models lstms machine-learning nanodegree neural-network nlp pytorch recurrent-neural-networks rnn sentiment-analysis style-transfer transfer-learning udacity udacity-nanodegree

Last synced: 07 Aug 2024

https://github.com/MinkaiXu/ConfVAE-ICML21

An End-to-End Framework for Molecular Conformation Generation via Bilevel Programming (ICML'21)

bilevel-optimization computational-biology computational-chemistry conformation generative-models graph-neural-networks icml-2021 molecule pytorch vae

Last synced: 08 Aug 2024

https://github.com/IBM/MAX-Fast-Neural-Style-Transfer

Generate a new image that mixes the content of a source image with the style of another image.

docker-image machine-learning machine-learning-models pytorch

Last synced: 04 Aug 2024

https://github.com/astorfi/cor-gan

:unlock: COR-GAN: Correlation-Capturing Convolutional Neural Networks for Generating Synthetic Healthcare Records

deep-learning generative-adversarial-network healthcare python pytorch synthetic-dataset-generation

Last synced: 02 Aug 2024

https://github.com/bage79/nlp4kor

Natural Language Processing for Korean with Deep Learning

deep-learning korean natural-language-processing pytorch tensorflow word2vec

Last synced: 25 Aug 2024

https://github.com/antofuller/configaformers

A python library for highly configurable transformers - easing model architecture search and experimentation.

artificial-intelligence deep-learning pytorch transformers

Last synced: 03 Aug 2024

https://github.com/Time0o/colorful-colorization

A from-scratch PyTorch implementation of "Colorful Image Colorization" by Zhang et al.

convolutional-neural-networks deep-learning image-colorization pytorch

Last synced: 03 Aug 2024

https://github.com/iamalexkorotin/Wasserstein2GenerativeNetworks

PyTorch implementation of "Wasserstein-2 Generative Networks" (ICLR 2021)

deep-learning gan generative-adversarial-network optimal-transport pytorch style-transfer

Last synced: 31 Jul 2024

https://github.com/MR-HosseinzadehTaher/BenchmarkTransferLearning

Official PyTorch Implementation and Pre-trained Models for Benchmarking Transfer Learning for Medical Image Analysis

benchmarking deep-learning domain-adaptation imagenet-models inaturalist-2021 medical-imaging pytorch self-supervised-learning supervised-learning transfer-learning

Last synced: 01 Aug 2024

https://github.com/hristo-vrigazov/dnn.cool

A framework for multi-task learning, where you may precondition tasks and compose them into bigger tasks. Conditional objectives and per-task evaluations and interpretations.

catalyst deep-learning deep-neural-network deep-neural-networks multitask-learning multitasking pytorch task-composition task-precondition

Last synced: 03 Aug 2024

https://github.com/pyronear/pyro-vision

Computer vision library for wildfire detection 🌲 Deep learning models in PyTorch & ONNX for inference on edge devices (e.g. Raspberry Pi)

computer-vision deep-learning image-classification keypoint-detection object-detection onnx python pytorch wildfire

Last synced: 08 Aug 2024

https://github.com/FerdinandZhong/punctuator

A small seq2seq punctuator tool based on DistilBERT

bert bert-ner chinese-nlp deep-learning nlp punctuation pytorch seq2seq

Last synced: 03 Aug 2024

https://github.com/qq456cvb/CPPF

CPPF: Towards Robust Category-Level 9D Pose Estimation in the Wild (CVPR2022)

cvpr2022 deep-learning pose-estimation pytorch

Last synced: 31 Jul 2024

https://github.com/lonePatient/bert-sentence-similarity-pytorch

This repo contains a PyTorch implementation of a pretrained BERT model for sentence similarity task.

bert nlp pytorch sentence-similarity text-classification

Last synced: 01 Aug 2024

https://github.com/j-min/VPGen

Visual Programming for Text-to-Image Generation and Evaluation (NeurIPS 2023)

pytorch step-by-step text-to-image text-to-image-evaluation text-to-image-generation

Last synced: 31 Jul 2024

https://github.com/lingyunwu14/STFT

Spatial-Temporal Feature Transformation for Video Object Detection, MICCAI2021

pytorch video-object-detection

Last synced: 02 Aug 2024

https://github.com/deepsense-ai/Keras-PyTorch-AvP-transfer-learning

We pit Keras and PyTorch against each other, showing their strengths and weaknesses in action. We present a real problem, a matter of life-and-death: distinguishing Aliens from Predators!

alien keras keras-classification-models keras-models keras-neural-networks keras-tutorials predator pytorch pytorch-cnn pytorch-implementation pytorch-tutorial pytorch-tutorials

Last synced: 07 Aug 2024

https://github.com/GilesStrong/lumin

LUMIN - a deep learning and data science ecosystem for high-energy physics.

deep-learning hep machine-learning physics pytorch science statistics

Last synced: 05 Aug 2024

https://github.com/qq456cvb/CanonicalVoting

Canonical Voting: Towards Robust Oriented Bounding Box Detection in 3D Scenes (CVPR2022)

3d-object-detection 3d-scenes cvpr2022 deep-learning hough-transform pytorch

Last synced: 31 Jul 2024

https://github.com/khakhulin/compressed-transformer

Compression of NMT transformer model with tensor methods

compression deep-learning mnist nlp nmt pytorch tensor-train transformer translation tucker

Last synced: 04 Aug 2024

https://github.com/arched-dev/mapextrackt

Pytorch Feature Map Extractor

feature-extraction neural-network pytorch

Last synced: 03 Aug 2024

https://github.com/kamo-naoyuki/pytorch_complex

A temporal module for PyTorch-ComplexTensor

complex-numbers dnn python python3 pytorch

Last synced: 01 Aug 2024

https://github.com/JosephKJ/ELI

(CVPR 2022) Energy-based Latent Aligner for Incremental Learning

continual-learning cvpr cvpr2022 ebm energy-based-model incremental-learning latent-aligning pytorch

Last synced: 03 Aug 2024

PyTorch Awesome Lists
Awesome-pytorch-list 704 awesome-semantic-segmentation-pytorch 35 Awesome-pytorch-list-CNVersion 692 CV-pretrained-model 103 awesome-colab-notebooks 502 awesome-AutoML-and-Lightweight-Models 150 Awesome-Denoise 164 ai4artists 286 arctic 34 awesome-yolov8-models 43 awesome-cuda-and-hpc 424 awesome-python-machine-learning-resources 1,884 awesome-adaptive-computation 171 awesome-NILM-with-code 47 awesome-dotnet-machine-learning 134 awesome-Multi-Document-Summarization 56 PyTorch-Guide 616 awesome-fastai 42 awesome-ai-tools 499
PyTorch Categories
Pytorch & related libraries 228 Pytorch & related libraries｜Pytorch & 相关库 218 Research 197 Tutorials 192 文本数据和NLP 182 Blogs 164 Uncategorized 161 Frameworks 142 数据可视化 130 机器学习框架 123 1.) Neural Architecture Search 113 图像数据与CV 105 Learning 102 Blogposts :sparkle: 101 模型的可解释性 100 Interactive Learning & Tools 96 超参数优化和AutoML 92 Tutorials, books, & examples 82 工作流程和实验跟踪 80 Tutorials & books & examples｜教程 & 书籍 & 示例 80