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/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: 13 Nov 2024

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

PyTorch implementation of SIFT descriptor

cnn descriptor image-matching local-features pytorch sift

Last synced: 01 Nov 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 Oct 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/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: 28 Oct 2024

https://github.com/markdtw/vqa-winner-cvprw-2017

Pytorch implementation of winner from VQA Chllange Workshop in CVPR'17

pytorch visual-question-answering

Last synced: 13 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: 29 Oct 2024

https://github.com/jayleicn/TVQA

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

dataset pytorch tvqa videoqa

Last synced: 07 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: 07 Nov 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: 05 Nov 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/wkcn/MobulaOP

A Simple & Flexible Cross Framework Operators Toolkit

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

Last synced: 08 Nov 2024

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

Batch generation from xarray datasets

keras pytorch tensorflow xarray xarray-accessor

Last synced: 22 Oct 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/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/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: 08 Nov 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: 09 Nov 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/huggingface/diffusion-fast

Faster generation with text-to-image diffusion models.

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

Last synced: 04 Nov 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/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: 28 Oct 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/isLinXu/YOLOv8_Efficient

🚀Simple and efficient use for Ultralytics yolov8🚀

deeplearning pytorch ultralytics yolov8

Last synced: 09 Nov 2024

https://github.com/nvidia/modulus-sym

Framework providing pythonic APIs, algorithms and utilities to be used with Modulus core to physics inform model training as well as higher level abstraction for domain experts

deep-learning machine-learning nvidia-gpu physics pytorch

Last synced: 29 Oct 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/shinianzhihou/ChangeDetection

A framework for change detection using PyTorch

change-detection pytorch remote-sensing siamese unet

Last synced: 03 Aug 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: 12 Nov 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: 13 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/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/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/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/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/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/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/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/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/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/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/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/floydhub/dockerfiles

Deep Learning Dockerfiles

deep-learning docker pytorch tensorflow torch

Last synced: 06 Nov 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/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/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: 13 Nov 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/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/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/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/nvidia/torchfort

An Online Deep Learning Interface for HPC programs on NVIDIA GPUs

deep-learning fortran libtorch pytorch

Last synced: 13 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/lxxue/frnn

Fixed Radius Nearest Neighbor Search on GPU

cuda nearest-neighbor-search pytorch

Last synced: 11 Nov 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/jingxil/Neural-Decision-Forests

An implementation of the Deep Neural Decision Forests in PyTorch

pytorch

Last synced: 30 Oct 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/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/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/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/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/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/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: 13 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/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/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: 13 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/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/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/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/xalanq/chinese-sentiment-classification

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

nlp pytorch

Last synced: 07 Nov 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/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