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/hejingwenhejingwen/adafm

CVPR2019 (oral) Modulating Image Restoration with Continual Levels via Adaptive Feature Modification Layers (AdaFM). PyTorch implementation

adafm cvpr2019 cvpr2019-oral denoising image-restoration low-level-vision pytorch super-resolution

Last synced: 21 Nov 2024

https://github.com/SurajDonthi/Multi-Camera-Person-Re-Identification

State-of-the-art model for person re-identification in Multi-camera Multi-Target Tracking. Benchmarked on Market-1501 and DukeMTMTC-reID datasets.

computer-vision convolutional-neural-networks deep-learning dukemtmc-reid market-1501 mtmct multi-camera-tracking person-reidentification pytorch pytorch-lightning spatial-temporal

Last synced: 28 Oct 2024

https://github.com/lxxue/frnn

Fixed Radius Nearest Neighbor Search on GPU

cuda nearest-neighbor-search pytorch

Last synced: 18 Feb 2025

https://github.com/iamyuanchung/Autoregressive-Predictive-Coding

Autoregressive Predictive Coding: An unsupervised autoregressive model for speech representation learning

pre-training pytorch representation-learning self-supervised-learning unsupervised-learning

Last synced: 27 Nov 2024

https://github.com/thnkim/OpenFacePytorch

PyTorch module to use OpenFace's nn4.small2.v1.t7 model

openface pytorch

Last synced: 14 Nov 2024

https://github.com/guanyingc/sdps-net

Learning Based Uncalibrated Photometric Stereo for Non-Lambertian Surface (CVPR 2019 Oral)

deep-learning non-lambertian photometric-stereo pytorch uncalibrated

Last synced: 18 Feb 2025

https://github.com/Audio-WestlakeU/NBSS

The official repo of NBC & SpatialNet for multichannel speech separation, denoising, and dereverberation

denoising dereverberation enhancement full-band multi-channel narrow-band pytorch separation speech

Last synced: 28 Oct 2024

https://github.com/AlanLi1997/slim-neck-by-gsconv

Pytorch implementation of the 'Slim-neck by GSConv: a lightweight-design for real-time detector architectures'

gsconv lightweight-framework objectdetection pytorch yolo yolov4 yolov5

Last synced: 09 Nov 2024

https://github.com/bat67/pytorch-fcn-easiest-demo

PyTorch Implementation of Fully Convolutional Networks (a very simple and easy demo).

cnn computer-vision fcn pytorch pytorch-implementation pytorch-implmention semantic-segmentation

Last synced: 20 Nov 2024

https://github.com/htdt/hyp_metric

Hyperbolic Vision Transformers: Combining Improvements in Metric Learning | Official repository

computer-vision machine-learning pytorch

Last synced: 15 Nov 2024

https://github.com/mblondel/fenchel-young-losses

Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses

loss-functions probabilistic-classification pytorch sklearn tensorflow

Last synced: 19 Dec 2024

https://github.com/tbepler/topaz

Pipeline for particle picking in cryo-electron microscopy images using convolutional neural networks trained from positive and unlabeled examples. Also featuring micrograph and tomogram denoising with DNNs.

cnn cryo-em cryoem denoising machine-learning object-detection particle-picking positive-unlabeled pytorch topaz topaz-denoise topaz-em

Last synced: 16 Feb 2025

https://github.com/willbrennan/semanticsegmentation

A framework for training segmentation models in pytorch on labelme annotations with pretrained examples of skin, cat, and pizza topping segmentation

birds bisenet bisenetv2 cats coco computer-vision labelme labelme-annotations pizza pizza-toppings pytorch segmentation semantic-segmentation skin-detection skin-segmentation torchvision

Last synced: 17 Nov 2024

https://github.com/belskikh/kekas

Just another DL library

deep-learning kek neural-networks pytorch

Last synced: 15 Nov 2024

https://github.com/suinleelab/path_explain

A repository for explaining feature attributions and feature interactions in deep neural networks.

explainable-ai interpretable-deep-learning machine-learning pytorch tensorflow

Last synced: 11 Nov 2024

https://github.com/drsleep/refinenet-pytorch

RefineNet-101 VOC in PyTorch

pytorch refinenet semantic-segmentation

Last synced: 16 Nov 2024

https://github.com/snap-research/cat

[CVPR 2021] Teachers Do More Than Teach: Compressing Image-to-Image Models (CAT)

compression cyclegan deep-learning gan gaugan image-to-image pix2pix pytorch

Last synced: 20 Dec 2024

https://github.com/ben-louis/deep-image-analogy-pytorch

A python implementation of Deep-Image-Analogy based on pytorch.

pytorch style-transfer

Last synced: 19 Dec 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: 19 Dec 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: 05 Nov 2024

https://github.com/eriklindernoren/action-recognition

Exploration of different solutions to action recognition in video, using neural networks implemented in PyTorch.

action-recognition pytorch video-classification

Last synced: 11 Nov 2024

https://github.com/patball1/detectree2

Python package for automatic tree crown delineation based on the Detectron2 implementation of Mask R-CNN

deep-learning detectron2 python pytorch

Last synced: 17 Feb 2025

https://github.com/theonesud/Pytorch-Model-Zoo

A collection of deep learning models implemented in PyTorch

collection deep-learning models neural-networks pytorch

Last synced: 27 Nov 2024

https://github.com/ChrisWaites/pyvacy

Differentially Private Optimization for PyTorch 👁🙅‍♀️

differential-privacy pytorch

Last synced: 02 Nov 2024

https://github.com/BMW-InnovationLab/BMW-Anonymization-API?referral=top-free-anonymization-tools-apis-and-open-source-models

This repository allows you to anonymize sensitive information in images/videos. The solution is fully compatible with the DL-based training/inference solutions that we already published/will publish for Object Detection and Semantic Segmentation.

anonymization-api anonymization-technique bmw computer-vision data-anonymization deep-learning docker image-transformation inference no-code object-detection openvino privacy-enhancing-technologies privacy-protection pytorch semantic-segmentation tensorflow tensorflow-training video video-anonymization

Last synced: 02 Dec 2024

https://github.com/wimlds-trojmiasto/detect-waste

AI to Combat Environmental Pollution - detecting plastic waste in the environment to combat environmental pollution and promote circular economy (Deep Learning, PyTorch)

cnn deep-learning detr efficientdet fastrcnn litter maskrcnn neural-networks object-detection python pytorch trash waste-detection

Last synced: 14 Nov 2024

https://github.com/snap-research/CAT

[CVPR 2021] Teachers Do More Than Teach: Compressing Image-to-Image Models (CAT)

compression cyclegan deep-learning gan gaugan image-to-image pix2pix pytorch

Last synced: 28 Nov 2024

https://github.com/czczup/urst

[AAAI 2022] Towards Ultra-Resolution Neural Style Transfer via Thumbnail Instance Normalization

neural-style-transfer pytorch

Last synced: 19 Dec 2024

https://github.com/tjuskyzhang/Scaled-YOLOv4-TensorRT

Got 100fps on TX2. Got 500fps on GeForce GTX 1660 Ti. If the project is useful to you, please Star it.

pytorch scaled-yolov4 scaled-yolov4-tenosrrt tensorrt tx2 yolo yolov4-csp yolov4-csp-tensorrt yolov4-large yolov4-large-tensorrt yolov4-tiny yolov4-tiny-tensorrt

Last synced: 09 Nov 2024

https://github.com/nasdin/reinforcementlearning-atarigame

Pytorch LSTM RNN for reinforcement learning to play Atari games from OpenAI Universe. We also use Google Deep Mind's Asynchronous Advantage Actor-Critic (A3C) Algorithm. This is much superior and efficient than DQN and obsoletes it. Can play on many games

a3c a3c-lstm actor-critic adam asynchronous-advantage-actor-critic deep-reinforcement-learning lstm openai-gym python pytorch reinforcement-agents reinforcement-learning reinforcement-learning-algorithms rmsprop universe

Last synced: 15 Feb 2025

https://github.com/BMW-InnovationLab/BMW-Anonymization-API

This repository allows you to anonymize sensitive information in images/videos. The solution is fully compatible with the DL-based training/inference solutions that we already published/will publish for Object Detection and Semantic Segmentation.

anonymization-api anonymization-technique bmw computer-vision data-anonymization deep-learning docker image-transformation inference no-code object-detection openvino privacy-enhancing-technologies privacy-protection pytorch semantic-segmentation tensorflow tensorflow-training video video-anonymization

Last synced: 05 Nov 2024

https://github.com/fedml-ai/fedgraphnn

FedGraphNN: A Federated Learning Platform for Graph Neural Networks with MLOps Support. The previous research version is accepted to ICLR'2021 - DPML and MLSys'21 - GNNSys workshops.

deep-learning distributed-learning federated-learning federated-learning-framework fedml gnns graph-neural-networks machine-learning pytorch tensorflow

Last synced: 18 Feb 2025

https://github.com/markdtw/meta-learning-lstm-pytorch

pytorch implementation of Optimization as a Model for Few-shot Learning

few-shot-learning learning-to-learn meta-learning pytorch

Last synced: 19 Dec 2024

https://github.com/cure-lab/deciwatch

[ECCV 2022] Official implementation of the paper "DeciWatch: A Simple Baseline for 10x Efficient 2D and 3D Pose Estimation"

2d-human-pose 3d-body-recovery 3d-pose-estimation body-reconstruction deep-learning eccv eccv2022 efficiency efficient-inference efficient-neural-networks human-pose-estimation pose-estimation pytorch

Last synced: 20 Dec 2024

https://github.com/david-knigge/ccnn

Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/abs/2301.10540.

convolutional-neural-networks deep-learning pytorch

Last synced: 28 Oct 2024

https://github.com/phamquiluan/publaynet

ICDAR 2019: MaskRCNN on PubLayNet datasets. Paragraph detection, table detection, figure detection,...

document-layout-analysis figure-detection mask-rcnn object-detection paragraph-detection pretrained-models publaynet pytorch table-detection

Last synced: 17 Dec 2024

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

https://github.com/HennyJie/BrainGB

Officially Accepted to IEEE Transactions on Medical Imaging (TMI, IF: 11.037) - Special Issue on Geometric Deep Learning in Medical Imaging.

ai4health brain-networks gcn graph-convolutional-networks graph-neural-networks network-embedding pytorch

Last synced: 12 Nov 2024

https://github.com/shi-labs/fcf-inpainting

[WACV 2023] Keys to Better Image Inpainting: Structure and Texture Go Hand in Hand

fast-fourier-convolution fcfgan image-inpainting pytorch stylegan2

Last synced: 19 Dec 2024

https://github.com/kennardwang/vtuber-momosehiyori

😀 Live2D VTuber, made by Unity and Cubism. This project is based on TCP socket, OpenCV and deep learning

csharp cubism deep-learning dlib live2d opencv pycharm python3 pytorch tcp-socket unity vtuber

Last synced: 10 Jan 2025

https://github.com/d-li14/psconv

[ECCV 2020] PSConv: Squeezing Feature Pyramid into One Compact Poly-Scale Convolutional Layer

convolution eccv2020 feature-pyramids instance-segmentation mmdetection multi-scale object-detection pytorch

Last synced: 07 Nov 2024

https://github.com/phamquiluan/PubLayNet

ICDAR 2019: MaskRCNN on PubLayNet datasets. Paragraph detection, table detection, figure detection,...

document-layout-analysis figure-detection mask-rcnn object-detection paragraph-detection pretrained-models publaynet pytorch table-detection

Last synced: 28 Nov 2024

https://github.com/acecoooool/dss-pytorch

:star: PyTorch implement of Deeply Supervised Salient Object Detection with Short Connection

python pytorch salient-object-detection

Last synced: 09 Nov 2024

https://github.com/pqhieu/jsis3d

[CVPR'19] JSIS3D: Joint Semantic-Instance Segmentation of 3D Point Clouds

cvpr deep-learning instance-segmentation point-cloud pytorch

Last synced: 28 Oct 2024

https://github.com/VSainteuf/pytorch-psetae

PyTorch implementation of the model presented in "Satellite Image Time Series Classification with Pixel-Set Encoders and Temporal Self-Attention"

agriculture computer-vision cvpr cvpr2020 deep-learning earth-observation machine-learning pytorch remote-sensing satellite-image self-attention spatio-temporal time-series-classification transformer-architecture

Last synced: 15 Nov 2024

https://github.com/csyanbin/TPN-pytorch

Pytorch Code for ICLR19 paper: Learning to Propagate Labels: Transductive Propagation Network for Few-shot Learning.

few-shot-learning label-propagation meta-learning pytorch transductive-few-shot-learning

Last synced: 26 Oct 2024

https://github.com/eriklindernoren/Action-Recognition

Exploration of different solutions to action recognition in video, using neural networks implemented in PyTorch.

action-recognition pytorch video-classification

Last synced: 06 Nov 2024

https://github.com/Chris-hughes10/pytorch-accelerated

A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop which is flexible enough to handle the majority of use cases, and capable of utilizing different hardware options with no code changes required. Docs: https://pytorch-accelerated.readthedocs.io/en/latest/

deep-learning huggingface-accelerate huggingface-examples pytorch pytorch-tutorial timm-examples training

Last synced: 23 Nov 2024

https://github.com/xiaoyu258/GeoProj

Blind Geometric Distortion Correction on Images Through Deep Learning

deep-learning distortion-correction pytorch

Last synced: 25 Oct 2024

https://github.com/d-li14/PSConv

[ECCV 2020] PSConv: Squeezing Feature Pyramid into One Compact Poly-Scale Convolutional Layer

convolution eccv2020 feature-pyramids instance-segmentation mmdetection multi-scale object-detection pytorch

Last synced: 14 Nov 2024

https://github.com/rusty1s/pytorch_spline_conv

Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch

geometric-deep-learning graph-neural-networks pytorch spline-cnn

Last synced: 14 Feb 2025

https://github.com/deepjavalibrary/djl-serving

A universal scalable machine learning model deployment solution

deep-learning deployment djl inference pytorch serving

Last synced: 19 Feb 2025

https://github.com/matajoh/fourier_feature_nets

Supplemental learning materials for "Fourier Feature Networks and Neural Volume Rendering"

computer-graphics computer-vision deep-learning pytorch

Last synced: 19 Dec 2024

https://github.com/dengbocong/text-similarity

文本相似度(匹配)计算,提供Baseline、训练、推理、指标分析...代码包含TensorFlow/Pytorch双版本

bert deep-learning mechine-learing model nlp pytorch similarity text-classification transformer

Last synced: 17 Feb 2025

https://github.com/clovaai/rebias

Official Pytorch implementation of ReBias (Learning De-biased Representations with Biased Representations), ICML 2020

bias-generalisation bias-generalization deep-learning icml icml-2020 icml2020 machine-learning pytorch rebias

Last synced: 18 Feb 2025

https://github.com/jzlianglu/pykaldi2

Yet another speech toolkit based on Kaldi and PyTorch

horovod kaldi pykaldi pytorch speech-toolkit

Last synced: 09 Dec 2024

https://github.com/Pilhyeon/BaSNet-pytorch

Official Pytorch Implementation of 'Background Suppression Network for Weakly-supervised Temporal Action Localization' (AAAI-20 Spotlight)

background-modeling deep-learning pytorch temporal-action-localization weakly-supervised-learning

Last synced: 13 Nov 2024

https://github.com/sangyx/d2l-torch

《动手学深度学习》 PyTorch 版本

chinese deep-learning jupyter-notebook python pytorch

Last synced: 17 Feb 2025

https://github.com/kangyeolk/Paint-by-Sketch

Stable Diffusion-based image manipulation method with a sketch and reference image

computer-vision diffusion-models image-editing image-generation image-manipulation pytorch stable-diffusion

Last synced: 31 Oct 2024

https://github.com/Js-Mim/mss_pytorch

Singing Voice Separation via Recurrent Inference and Skip-Filtering Connections - PyTorch Implementation. Demo:

deep-learning mss-pytorch music-source-separation pytorch recurrent-neural-networks skip-filtering-connections

Last synced: 14 Nov 2024

https://github.com/adobe-research/sam_inversion

[CVPR 2022] GAN inversion and editing with spatially-adaptive multiple latent layers

computer-graphics computer-vision deep-learning gan generative-adversarial-network image-manipulation machine-learning pytorch

Last synced: 20 Dec 2024

https://github.com/spytensor/detect_steel_bar

CCFDF AI 数钢筋大赛

detection gangjin pytorch retinanet

Last synced: 20 Dec 2024

https://github.com/mithril-security/bastionlab

A simple framework for privacy-friendly data science collaboration

deep-learning eda multi-party privacy pytorch secure-enclave

Last synced: 16 Feb 2025

https://github.com/xarray-contrib/xbatcher

Batch generation from xarray datasets

keras pytorch tensorflow xarray xarray-accessor

Last synced: 16 Feb 2025

https://github.com/ahirner/pytorch-retraining

Transfer Learning Shootout for PyTorch's model zoo (torchvision)

benchmark pytorch transfer-learning

Last synced: 15 Nov 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: 19 Feb 2025

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

https://github.com/batra-mlp-lab/visdial-rl

PyTorch code for Learning Cooperative Visual Dialog Agents using Deep Reinforcement Learning

computer-vision deep-learning natural-language-processing pytorch visual-dialog

Last synced: 27 Nov 2024

https://github.com/prrao87/fine-grained-sentiment

A comparison and discussion of different NLP methods for 5-class sentiment classification on the SST-5 dataset.

fasttext flair nlp python pytorch sentiment-analysis text-classification transformers

Last synced: 19 Feb 2025

https://github.com/nadeemlab/DeepLIIF

Deep Learning Inferred Multiplex ImmunoFluorescence for IHC Image Quantification (https://deepliif.org) [Nature Machine Intelligence'22, CVPR'22, MICCAI'23, Histopathology'23, MICCAI'24]

ai-ready cell-biology cell-segmentation cloud-native-applications datasets deep-learning digital-pathology image-segmentation immunohistochemical-images immunohistochemistry multiplex multitask-learning pathology pathology-gan pathology-image pytorch segmentation stitching super-resolution whole-slide-imaging

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

https://github.com/IlyaGusev/summarus

Models for automatic abstractive summarization

deep-learning machine-learning nlp pytorch summarization

Last synced: 04 Nov 2024