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/amanchadha/iseebetter

iSeeBetter: Spatio-Temporal Video Super Resolution using Recurrent-Generative Back-Projection Networks | Python3 | PyTorch | GANs | CNNs | ResNets | RNNs | Published in Springer Journal of Computational Visual Media, September 2020, Tsinghua University Press

cnn convolutional-neural-network convolutional-neural-networks gan gans generative-adversarial-network psnr pytorch recurrent-neural-networks residual-networks resnet resnets rnn self-supervised-learning srgan ssim super-resolution unsupervised-learning video-processing video-super-resolution

Last synced: 02 Feb 2025

https://github.com/ShirAmir/dino-vit-features

Official implementation for the paper "Deep ViT Features as Dense Visual Descriptors".

co-segmentation computer-vision deep-learning dino part-segmentation pytorch semantic-correspondence vision-transformers

Last synced: 04 Nov 2024

https://github.com/HikariTJU/LD

Localization Distillation for Object Detection (CVPR 2022, TPAMI 2023)

deep-learning detection knowledge-distillation ld object-detection pytorch

Last synced: 05 Nov 2024

https://github.com/official-stockfish/nnue-pytorch

Stockfish NNUE (Chess evaluation) trainer in Pytorch

chess deep-learning pytorch stockfish

Last synced: 01 Feb 2025

https://github.com/KevinMusgrave/pytorch-adapt

Domain adaptation made easy. Fully featured, modular, and customizable.

computer-vision deep-learning domain-adaptation machine-learning pytorch transfer-learning

Last synced: 05 Nov 2024

https://github.com/kakaoenterprise/JORLDY

Repository for Open Source Reinforcement Learning Framework JORLDY

framework gym mlagents pytorch reinforcement-learning

Last synced: 28 Nov 2024

https://github.com/QUVA-Lab/escnn

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

deep-learning equivariance geometric-deep-learning geometry invariant isometries pytorch rotations symmetries

Last synced: 13 Nov 2024

https://github.com/uber-research/deeppruner

DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)

iccv2019 patchmatch pytorch real-time stereo-matching stereo-vision

Last synced: 30 Jan 2025

https://github.com/kefirski/pytorch_RVAE

Recurrent Variational Autoencoder that generates sequential data implemented with pytorch

deep-learning nlp python pytorch vae

Last synced: 02 Nov 2024

https://github.com/quantylab/rltrader

파이썬과 케라스를 이용한 딥러닝/강화학습 주식투자 - 퀀트 투자, 알고리즘 트레이딩을 위한 최첨단 해법 입문 (개정판)

deep-learning keras python pytorch reinforcement-learning tensorflow

Last synced: 31 Jan 2025

https://github.com/quva-lab/escnn

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

deep-learning equivariance geometric-deep-learning geometry invariant isometries pytorch rotations symmetries

Last synced: 31 Jan 2025

https://github.com/ymcui/PERT

PERT: Pre-training BERT with Permuted Language Model

bert nlp plm pre-trained-model pytorch tensorflow transformers

Last synced: 16 Nov 2024

https://github.com/seeed-studio/modelassistant

Seeed SenseCraft Model Assistant is an open-source project focused on embedded AI. 🔥🔥🔥

arduino deep-learning esp32 image-classification jetson ncnn object-detection onnx openmmlab pytorch raspberry-pi tflite tinyml yolov5

Last synced: 05 Feb 2025

https://github.com/ymcui/pert

PERT: Pre-training BERT with Permuted Language Model

bert nlp plm pre-trained-model pytorch tensorflow transformers

Last synced: 06 Feb 2025

https://github.com/fendouai/pytorch1.0-cn

PyTorch 1.0 官方文档 中文版,欢迎关注微信公众号:磐创AI

pytorch pytorch-nlp pytorch-tutorial

Last synced: 30 Jan 2025

https://github.com/utkuozbulak/pytorch-cnn-adversarial-attacks

Pytorch implementation of convolutional neural network adversarial attack techniques

adversarial-example fast-gradient-sign fooling-images gradient-ascent pytorch pytorch-tutorial

Last synced: 30 Jan 2025

https://github.com/ybabakhin/kaggle_salt_bes_phalanx

Winning solution for the Kaggle TGS Salt Identification Challenge.

deep-learning image-segmentation keras pseudo-labeling pytorch semantic-segmentation u-net

Last synced: 30 Jan 2025

https://github.com/maum-ai/hififace

Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)

face-swapping pytorch pytorch-lightning wandb

Last synced: 05 Feb 2025

https://github.com/RLE-Foundation/RLeXplore

RLeXplore provides stable baselines of exploration methods in reinforcement learning, such as intrinsic curiosity module (ICM), random network distillation (RND) and rewarding impact-driven exploration (RIDE).

baselines efficient-algorithm exploration-strategy gym machine-learning pybullet pytorch reinforcement-learning robotics toolbox

Last synced: 04 Dec 2024

https://github.com/stared/thinking-in-tensors-writing-in-pytorch

Thinking in tensors, writing in PyTorch (a hands-on deep learning intro)

deep-learning pytorch pytorch-tutorial tensor

Last synced: 04 Feb 2025

https://github.com/francescosaveriozuppichini/pytorch-deep-learning-template

A Pytorch Computer Vision template to quick start your next project! 🚀🚀

computer-vision deep-learning python pytorch

Last synced: 02 Feb 2025

https://github.com/square/pysurvival

Open source package for Survival Analysis modeling

deep-learning machine-learning numpy python pytorch survival-analysis

Last synced: 12 Nov 2024

https://github.com/dvl-tum/mot_neural_solver

Official PyTorch implementation of "Learning a Neural Solver for Multiple Object Tracking" (CVPR 2020 Oral).

graph-neural-networks multi-object-tracking pytorch

Last synced: 28 Oct 2024

https://github.com/beomi/infinitransformer

Unofficial PyTorch/🤗Transformers(Gemma/Llama3) implementation of Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention

gemma huggingface infinitransformer llama llama3 pytorch transformers

Last synced: 04 Feb 2025

https://github.com/megvii-research/MOTRv2

[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors

end-to-end multi-object-tracking pytorch transformer

Last synced: 28 Oct 2024

https://github.com/uber-research/DeepPruner

DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)

iccv2019 patchmatch pytorch real-time stereo-matching stereo-vision

Last synced: 31 Oct 2024

https://github.com/Megvii-CSG/MegReader

A research project for text detection and recognition using PyTorch 1.2.

ctc deep-learning ocr pytorch text-detection text-detection-recognition text-recognition

Last synced: 04 Nov 2024

https://github.com/imoonlab/meshnet

MeshNet: Mesh Neural Network for 3D Shape Representation (AAAI 2019)

3d-shapes deep-learning mesh pytorch

Last synced: 03 Feb 2025

https://github.com/hzwer/cvpr2023-dmvfn

CVPR2023 (highlight) - A Dynamic Multi-Scale Voxel Flow Network for Video Prediction

computer-vision cvpr2023 deep-learning pytorch

Last synced: 03 Feb 2025

https://github.com/davidmascharka/tbd-nets

PyTorch implementation of "Transparency by Design: Closing the Gap Between Performance and Interpretability in Visual Reasoning"

deep-learning machine-learning neural-networks pytorch visual-question-answering visualization vqa

Last synced: 06 Feb 2025

https://github.com/sony/model_optimization

Model Compression Toolkit (MCT) is an open source project for neural network model optimization under efficient, constrained hardware. This project provides researchers, developers, and engineers advanced quantization and compression tools for deploying state-of-the-art neural networks.

deep-learning deep-neural-networks edge-ai machine-learning network-compression network-quantization neural-network optimizer ptq pytorch qat quantization tensorflow

Last synced: 01 Feb 2025

https://github.com/atgambardella/pytorch-es

Evolution Strategies in PyTorch

machinelearning python pytorch

Last synced: 02 Nov 2024

https://github.com/asyml/vision-transformer-pytorch

Pytorch version of Vision Transformer (ViT) with pretrained models. This is part of CASL (https://casl-project.github.io/) and ASYML project.

pytorch vision-transformer

Last synced: 30 Jan 2025

https://github.com/talebolano/yolov3-network-slimming

yolov3 network slimming剪枝的一种实现

network pruning pytorch slimming yolo

Last synced: 09 Nov 2024

https://github.com/nashory/delf-pytorch

PyTorch Implementation of "Large-Scale Image Retrieval with Attentive Deep Local Features"

image-retrieval local-features pytorch

Last synced: 30 Jan 2025

https://github.com/mir-group/allegro

Allegro is an open-source code for building highly scalable and accurate equivariant deep learning interatomic potentials

atomistic-simulations computational-chemistry deep-learning drug-discovery force-fields interatomic-potentials machine-learning materials-science molecular-dynamics pytorch

Last synced: 19 Nov 2024

https://github.com/IBM/aihwkit

IBM Analog Hardware Acceleration Kit

ai analog-devices cuda neural-networks pytorch

Last synced: 17 Nov 2024

https://github.com/justusschock/shapenet

PyTorch implementation of "Super-Realtime Facial Landmark Detection and Shape Fitting by Deep Regression of Shape Model Parameters" predicting facial landmarks with up to 400 FPS

deep-learning delira facial-keypoints facial-keypoints-cnn facial-landmarks landmark-detection pytorch shape shape-models shape-prediction shape-predictor

Last synced: 06 Feb 2025

https://github.com/bshall/hubert

HuBERT content encoders for: A Comparison of Discrete and Soft Speech Units for Improved Voice Conversion

pytorch representation-learning speech voice-conversion

Last synced: 03 Feb 2025

https://github.com/lambdalabsml/distributed-training-guide

Best practices & guides on how to write distributed pytorch training code

cluster cuda deepspeed distributed-training fsdp gpu gpu-cluster kuberentes lambdalabs mpi nccl pytorch sharding slurm

Last synced: 03 Feb 2025

https://github.com/vballoli/nfnets-pytorch

NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/

adaptive-gradient-clipping deepmind image-classification nfnets paper pytorch sgd sota

Last synced: 28 Oct 2024

https://github.com/ekzhang/fastseg

📸 PyTorch implementation of MobileNetV3 for real-time semantic segmentation, with pretrained weights & state-of-the-art performance

aspp cityscapes computer-vision deep-learning deeplabv3 edge-computing efficientnet kitti-dataset mapillary-vistas-dataset mobilenetv3 pytorch semantic-segmentation

Last synced: 31 Jan 2025

https://github.com/apple/ml-gmpi

[ECCV 2022, Oral Presentation] Official PyTorch implementation of GMPI

3d eccv2022 generative-adversarial-network generative-model multiplane-images neural-rendering pytorch

Last synced: 05 Feb 2025

https://github.com/microsoft/nn-meter

A DNN inference latency prediction toolkit for accurately modeling and predicting the latency on diverse edge devices.

deep-learning deep-neural-networks edge-ai edge-computing efficient-model inference latency machine-learning neural-architecture-search onnx-models python pytorch tensorflow-models

Last synced: 31 Jan 2025

https://github.com/inisis/brocolli

Everything in Torch Fx

caffe onnx pytorch quantization

Last synced: 01 Feb 2025

https://github.com/pytorch/torchx

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

airflow aws-batch components deep-learning distributed-training kubernetes machine-learning pipelines python pytorch ray slurm

Last synced: 30 Jan 2025

https://github.com/Pointcept/PointTransformerV2

[NeurIPS'22] An official PyTorch implementation of PTv2.

3d-vision point-cloud pytorch

Last synced: 28 Oct 2024

https://github.com/taishan1994/pytorch_triple_extraction

基于pytorch的中文三元组提取(命名实体识别+关系抽取)

bert named-entity-recognition pytorch relation-extraction triple-extraction

Last synced: 06 Feb 2025

https://github.com/kevinnan-teen/intelligent-traffic-based-on-cv

本项目是基于计算机视觉的端到端交通路口智能监控系统.采用的设计架构由SRS流媒体服务器、GPU服务器、Local客户端三部分组成.可将远端视频流通过rtmp协议传输到流媒体服务器,然后经过目标检测等一些列算法对视频进行分析,最后在本地客户端查看分析结果.项目主要用Python实现,流媒体服务器采用开源的SRS实时视频服务器搭建,GPU服务器使用YOLO模型实现道路目标如人、车、交通灯等物体的识别,同时还支持并发连接.客户端基于PyQt5进行开发.

python pytorch srs yolov3

Last synced: 06 Feb 2025

https://github.com/Algolzw/image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.

deblurring dehazing denoising deraining diffusion-models image-restoration inpainting pytorch score-based stochastic-differential-equations super-resolution

Last synced: 30 Oct 2024

https://github.com/ZhangGongjie/Meta-DETR

[T-PAMI 2022] Meta-DETR for Few-Shot Object Detection: Official PyTorch Implementation

computer-vision few-shot-object-detection meta-learning object-detection pytorch

Last synced: 15 Oct 2024

https://github.com/lucasalegre/morl-baselines

Multi-Objective Reinforcement Learning algorithms implementations.

gym gymnasium mo-gymnasium morl multi-objective multi-objective-rl pytorch reinforcement-learning rl rl-algorithms

Last synced: 01 Feb 2025

https://github.com/mjhydri/BeatNet

BeatNet is state-of-the-art (Real-Time) and Offline joint music beat, downbeat, tempo, and meter tracking system using CRNN and particle filtering. (ISMIR 2021's paper implementation).

beat-tracking beatnet crnn-network dnn-beat-tracking downbeat-tracking joint-model meter-detection online-time-signature-detection particle-filtering pytorch real-time real-time-beat-tracker real-time-downbeat-tracker real-time-tempo state-of-the-art

Last synced: 22 Nov 2024

https://github.com/wpeebles/g.pt

Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"

deep-learning diffusion-models pytorch transformers

Last synced: 30 Jan 2025

https://github.com/golsun/DialogRPT

EMNLP 2020: "Dialogue Response Ranking Training with Large-Scale Human Feedback Data"

conversational-ai dataset dialog dialog-datasets dialogpt gpt-2 human-feedback-data pretrained-models pytorch transformers

Last synced: 30 Oct 2024

https://github.com/lukecavabarrett/pna

Implementation of Principal Neighbourhood Aggregation for Graph Neural Networks in PyTorch, DGL and PyTorch Geometric

dgl graph-machine-learning graph-neural-networks pytorch pytorch-geometric

Last synced: 28 Nov 2024

https://github.com/v-iashin/SpecVQGAN

Source code for "Taming Visually Guided Sound Generation" (Oral at the BMVC 2021)

audio audio-generation bmvc evaluation-metrics gan melgan multi-modal pytorch transformer vas vggsound video video-features video-understanding vqvae

Last synced: 06 Nov 2024

https://github.com/SamsungLabs/adaptis

[ICCV19] AdaptIS: Adaptive Instance Selection Network, https://arxiv.org/abs/1909.07829

adaptis cityscapes instance-segmentation mapillary-vistas-dataset ms-coco mxnet panoptic-segmentation pytorch

Last synced: 08 Nov 2024

https://github.com/Junjue-Wang/LoveDA

[NeurIPS 2021] LoveDA: A Remote Sensing Land-Cover Dataset for Domain Adaptive Semantic Segmentation

domain-adaptation pytorch remote-sensing semantic-segmentation

Last synced: 15 Nov 2024

https://github.com/amirbar/detreg

Official implementation of the CVPR 2022 paper "DETReg: Unsupervised Pretraining with Region Priors for Object Detection".

deep-learning object-detection pytorch unsupervised-learning

Last synced: 05 Feb 2025

https://github.com/xplip/pixel

Research code for pixel-based encoders of language (PIXEL)

deep-learning deep-neural-networks language-model machine-learning nlp pytorch

Last synced: 05 Feb 2025

https://github.com/bshall/zerospeech

VQ-VAE for Acoustic Unit Discovery and Voice Conversion

acoustic-features pytorch speech-synthesis voice-conversion vq-vae zerospeech

Last synced: 06 Feb 2025

https://github.com/jacobgil/pytorch-explain-black-box

PyTorch implementation of Interpretable Explanations of Black Boxes by Meaningful Perturbation

blackbox deep-learning deeplearning pytorch visualization

Last synced: 06 Feb 2025

https://github.com/myungsub/cain

Source code for AAAI 2020 paper "Channel Attention Is All You Need for Video Frame Interpolation"

aaai2020 channel-attention dataset deep-convolutional-networks frame-interpolation pytorch video-frame-interpolation

Last synced: 06 Feb 2025

https://github.com/bethgelab/model-vs-human

Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)

benchmark pytorch robustness tensorflow toolbox

Last synced: 15 Nov 2024

https://github.com/domainadaptation/salad

A toolbox for domain adaptation and semi-supervised learning. Contributions welcome.

deep-learning domain-adaptation machine-learning pytorch

Last synced: 14 Nov 2024

https://github.com/BIGBALLON/PyTorch-CPP

PyTorch C++ inference with LibTorch

cpp demo imagenet inference libtorch opencv pytorch pytorch-cpp

Last synced: 27 Oct 2024

https://github.com/iMoonLab/MeshNet

MeshNet: Mesh Neural Network for 3D Shape Representation (AAAI 2019)

3d-shapes deep-learning mesh pytorch

Last synced: 28 Oct 2024

https://github.com/vita-group/sinnerf

[ECCV 2022] "SinNeRF: Training Neural Radiance Fields on Complex Scenes from a Single Image", Dejia Xu, Yifan Jiang, Peihao Wang, Zhiwen Fan, Humphrey Shi, Zhangyang Wang

gan nerf neural-radiance-fields pytorch pytorch-lightning

Last synced: 30 Jan 2025

https://github.com/szymonmaszke/torchdatasets

PyTorch dataset extended with map, cache etc. (tensorflow.data like)

cache concatenate dataset disk filter library map pytorch tensorflow tf-data torch

Last synced: 05 Feb 2025

https://github.com/hancyran/RepSurf

[CVPR 2022 Oral] Official implementation for "Surface Representation for Point Clouds"

cvpr2022 modelnet40 point-cloud pytorch s3dis scannet scanobjectnn sunrgbd

Last synced: 28 Oct 2024

https://github.com/rfeinman/pytorch-minimize

Newton and Quasi-Newton optimization with PyTorch

minimization newton-method optimization pytorch quasi-newton-method

Last synced: 03 Feb 2025

https://github.com/nimarb/pytorch_influence_functions

This is a PyTorch reimplementation of Influence Functions from the ICML2017 best paper: Understanding Black-box Predictions via Influence Functions by Pang Wei Koh and Percy Liang.

deep-learning influence-functions pytorch pytorch-implementation

Last synced: 01 Feb 2025

https://github.com/ecs-vlc/FMix

Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'

augmentation pytorch pytorch-lightning tensorflow torchbearer

Last synced: 13 Nov 2024

https://github.com/frgfm/holocron

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix

ademamix computer-vision cspdarknet53 darknet deep-learning object-detection pytorch resnet rexnet tridentnet unet-image-segmentation yolo yolov4

Last synced: 01 Feb 2025

https://github.com/yuanyuanli85/Fast_Human_Pose_Estimation_Pytorch

Pytorch Code for CVPR2019 paper "Fast Human Pose Estimation" https://arxiv.org/abs/1811.05419

knowledge-distillation pose-estimation pytorch stacked-hourglass-networks

Last synced: 05 Nov 2024

https://github.com/bknyaz/graph_nn

Graph Classification with Graph Convolutional Networks in PyTorch (NeurIPS 2018 Workshop)

gcn graph graph-classification graph-convolutional-networks graph-unet multigraph pooling pytorch

Last synced: 12 Nov 2024

https://github.com/liuruoze/mini-alphastar

(JAIR'2022) A mini-scale reproduction code of the AlphaStar program. Note: the original AlphaStar is the AI proposed by DeepMind to play StarCraft II. JAIR = Journal of Artificial Intelligence Research.

deep-learning deep-neural-networks deep-reinforcement-learning gaming imitation-learning mini-alphastar multi-agent-systems pytorch reinforcement-learning sc2replay self-playing-bot starcraft-ii-bot starcraft2 starcraft2-ai supervised-learning

Last synced: 06 Feb 2025