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/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: 30 Jul 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: 03 Aug 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: 31 Jul 2024

https://github.com/xiaoyu258/GeoProj

Blind Geometric Distortion Correction on Images Through Deep Learning

deep-learning distortion-correction pytorch

Last synced: 30 Jul 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: 01 Aug 2024

https://github.com/jzlianglu/pykaldi2

Yet another speech toolkit based on Kaldi and PyTorch

horovod kaldi pykaldi pytorch speech-toolkit

Last synced: 17 Aug 2024

https://github.com/recursionpharma/gflownet

GFlowNet library specialized for graph & molecular data

deep-learning gflownet graph-neural-network pytorch

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

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

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

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

benchmark pytorch transfer-learning

Last synced: 03 Aug 2024

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

https://github.com/pxiangwu/MotionNet

CVPR 2020, "MotionNet: Joint Perception and Motion Prediction for Autonomous Driving Based on Bird's Eye View Maps"

autonomous-driving cvpr2020 motion-prediction perception pytorch

Last synced: 31 Jul 2024

https://github.com/IlyaGusev/summarus

Models for automatic abstractive summarization

deep-learning machine-learning nlp pytorch summarization

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

https://github.com/wkcn/MobulaOP

A Simple & Flexible Cross Framework Operators Toolkit

cross-framework cupy deep-learning mxnet numpy operators pytorch scientific-computing

Last synced: 01 Aug 2024

https://github.com/fastaudio/fastaudio

🔊 Audio and fastai v2

audio deep-learning fastai gpu python pytorch

Last synced: 03 Aug 2024

https://github.com/j3soon/tbparse

Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow

machine-learning parser plot python pytorch reader tensorboard tensorboardx tensorflow

Last synced: 03 Aug 2024

https://github.com/ducha-aiki/pytorch-sift

PyTorch implementation of SIFT descriptor

cnn descriptor image-matching local-features pytorch sift

Last synced: 01 Aug 2024

https://github.com/AutoVision-cloud/SA-Det3D

[ICCVW-2021] SA-Det3D: Self-attention based Context-Aware 3D Object Detection

3d-deep-learning 3d-detection autonomous-driving object-detection point-cloud pytorch

Last synced: 31 Jul 2024

https://github.com/ShomyLiu/Neu-Review-Rec

A Toolkit for Neural Review-based Recommendation models with Pytorch.

daml deepconn narre pytorch recommendation-system review-based

Last synced: 03 Aug 2024

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

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

chinese deep-learning jupyter-notebook python pytorch

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

https://github.com/geekjr/quickai

QuickAI is a Python library that makes it extremely easy to experiment with state-of-the-art Machine Learning models.

ai artificial-intelligence bert deep-learning dl easy-to-use fast gpt gpt-neo huggingface-transformers ml neural-network nlp object-detection python pytorch quickai research tensorflow2 yolo

Last synced: 02 Aug 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: 31 Jul 2024

https://github.com/CleanPegasus/TensorDash

TensorDash is an application that lets you remotely monitor your deep learning model's metrics and notifies you when your model training is completed or crashed.

android deep-learning fastai java keras machine-learning pypi python python-package pytorch remote-monitoring tensorflow

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

https://github.com/SforAiDl/vformer

A modular PyTorch library for vision transformer models

pytorch vision-transformer

Last synced: 03 Aug 2024

https://github.com/JieShibo/PETL-ViT

[ICCV 2023] Binary Adapters, [AAAI 2023] FacT, [Tech report] Convpass

parameter-efficient-tuning pytorch transfer-learning visual-prompting

Last synced: 01 Aug 2024

https://github.com/jayleicn/TVQA

[EMNLP 2018] PyTorch code for TVQA: Localized, Compositional Video Question Answering

dataset pytorch tvqa videoqa

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

https://github.com/huggingface/diffusion-fast

Faster generation with text-to-image diffusion models.

diffusers diffusion-models pytorch sdxl text-to-image-generation

Last synced: 01 Aug 2024

https://github.com/uvipen/SSD-pytorch

SSD: Single Shot MultiBox Detector pytorch implementation focusing on simplicity

computer-vision deep-learning deep-neural-networks deeplearning detection detector neural-networks object-detection pytorch ssd ssdlite

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

https://github.com/IDSIA/modern-srwm

Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep RL Workshop) and "Accelerating Neural Self-Improvement via Bootstrapping" (ICLR 2023 Workshop)

fast-weight-programmers fast-weights pytorch reinforcement-learning rnn-based-few-shot-learning self-referential-learning self-referential-weight-matrix torchbeast transformers

Last synced: 02 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/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/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/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/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: 31 Jul 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: 01 Aug 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: 02 Aug 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 Jul 2024

https://github.com/floydhub/dockerfiles

Deep Learning Dockerfiles

deep-learning docker pytorch tensorflow torch

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

https://github.com/hou-yz/MVDet

[ECCV 2020] Codes and MultiviewX dataset for "Multiview Detection with Feature Perspective Transformation".

dataset detection detector multiview pedestrian-detection pytorch

Last synced: 31 Jul 2024

https://github.com/isLinXu/YOLOv8_Efficient

🚀Simple and efficient use for Ultralytics yolov8🚀

deeplearning pytorch ultralytics yolov8

Last synced: 02 Aug 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: 31 Jul 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/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: 02 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/jingxil/Neural-Decision-Forests

An implementation of the Deep Neural Decision Forests in PyTorch

pytorch

Last synced: 31 Jul 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/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: 03 Aug 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/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/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/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/kennethnym/infinifi

infinifi plays gentle lofi music in the background indefinitely

artificial-intelligence audiocraft lo-fi lo-fi-radio machine-learning musicgen pytorch

Last synced: 31 Jul 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/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: 02 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/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: 30 Jul 2024

https://github.com/kashif/firedup

Clone of OpenAI's Spinning Up in PyTorch

deep-learning pytorch reinforcement-learning spinningup

Last synced: 02 Aug 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/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/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: 30 Jul 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 Aug 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/aidos-lab/pytorch-topological

A topological machine learning framework based on PyTorch

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

Last synced: 02 Aug 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: 01 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: 31 Jul 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: 31 Jul 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