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/davidtvs/PyTorch-ENet

PyTorch implementation of ENet

camvid cityscape enet pytorch

Last synced: 01 Aug 2024

https://github.com/texttron/tevatron

Tevatron - A flexible toolkit for neural retrieval research and development.

dense-retrieval dpr flax information-retrieval jax pytorch question-answering transformer

Last synced: 02 Aug 2024

https://github.com/gordicaleksa/pytorch-GANs

My implementation of various GAN (generative adversarial networks) architectures like vanilla GAN (Goodfellow et al.), cGAN (Mirza et al.), DCGAN (Radford et al.), etc.

conditional-gan dc-gan deep-learning deep-learning-tutorial gan-tutorial gans generative-adversarial-nets generative-adversarial-networks machine-learning python pytorch vanilla-gan

Last synced: 02 Aug 2024

https://github.com/swz30/MIRNetv2

[TPAMI 2022] Learning Enriched Features for Fast Image Restoration and Enhancement. Results on Defocus Deblurring, Denoising, Super-resolution, and image enhancement

contrast-enhancement defocus-deblurring denoising high-resolution image-denoising image-enhancement image-processing image-restoration low-level-vision low-light-image-enhancement pytorch tpami

Last synced: 01 Aug 2024

https://github.com/SmallMunich/nutonomy_pointpillars

Convert pointpillars Pytorch Model To ONNX for TensorRT Inference

nutonomy onnx pointpillars pytorch tensorrt

Last synced: 31 Jul 2024

https://github.com/mir-aidj/all-in-one

All-In-One Music Structure Analyzer

allin1 beat-tracking music-structure-analysis pytorch

Last synced: 05 Aug 2024

https://github.com/mariusbrataas/flowpoints_ml

An intuitive approach to creating deep learning models

deep-learning flowchart js machine-learning python pytorch react tensorflow

Last synced: 03 Aug 2024

https://github.com/UM-ARM-Lab/pytorch_mppi

Model Predictive Path Integral (MPPI) with approximate dynamics implemented in pytorch

approximate-dynamics controls model-predictive-control mppi pytorch

Last synced: 31 Jul 2024

https://github.com/arthurdouillard/incremental_learning.pytorch

A collection of incremental learning paper implementations including PODNet (ECCV20) and Ghost (CVPR-W21).

continual-learning deep-learning incremental-learning lifelong-learning pytorch research

Last synced: 02 Aug 2024

https://github.com/mratsim/Amazon-Forest-Computer-Vision

Amazon Forest Computer Vision: Satellite Image tagging code using PyTorch / Keras with lots of PyTorch tricks

computer-vision data-augmentation deep-learning kaggle kaggle-competition keras neural-network-example neural-networks pytorch transfer-learning

Last synced: 03 Aug 2024

https://github.com/cezannec/capsule_net_pytorch

Readable implementation of a Capsule Network as described in "Dynamic Routing Between Capsules" [Hinton et. al.]

capsule-network pytorch pytorch-tutorial

Last synced: 02 Aug 2024

https://github.com/yifanzhang713/IA-SSD

Not All Points Are Equal: Learning Highly Efficient Point-based Detectors for 3D LiDAR Point Clouds (CVPR 2022, Oral)

3d-object-detection autonomous-driving ia-ssd object-detection point-cloud pytorch

Last synced: 31 Jul 2024

https://github.com/msaroufim/ml-design-patterns

Software Architecture for ML engineers

deep-learning design-patterns python pytorch systems

Last synced: 03 Aug 2024

https://github.com/itailang/SampleNet

Differentiable Point Cloud Sampling (CVPR 2020 Oral)

cvpr2020 deep-learning geometry-processing neural-network point-cloud pytorch sampling tensorlow

Last synced: 31 Jul 2024

https://github.com/felixgwu/img_classification_pk_pytorch

Quickly comparing your image classification models with the state-of-the-art models (such as DenseNet, ResNet, ...)

cifar computer-vision deep-learning pytorch

Last synced: 03 Aug 2024

https://github.com/Atcold/pytorch-CortexNet

PyTorch implementation of the CortexNet predictive model

deep-learning predictive-modeling pytorch self-supervised unsupervised-learning video

Last synced: 07 Aug 2024

https://github.com/Owen-Liuyuxuan/visualDet3D

Official Repo for Ground-aware Monocular 3D Object Detection for Autonomous Driving / YOLOStereo3D: A Step Back to 2D for Efficient Stereo 3D Detection

computer-vision icra monocular-3d-detection object-detection pytorch stereo-3d-detection

Last synced: 31 Jul 2024

https://github.com/DerryHub/BEVFormer_tensorrt

BEVFormer inference on TensorRT, including INT8 Quantization and Custom TensorRT Plugins (float/half/half2/int8).

bevformer cuda int8-inference pytorch quantization tensorrt-plugins

Last synced: 31 Jul 2024

https://github.com/wuyifan18/DeepLog

Pytorch Implementation of DeepLog.

anomaly-detection deeplearning pytorch sequence-prediction

Last synced: 01 Aug 2024

https://github.com/operatorai/modelstore

🏬 modelstore is a Python library that allows you to version, export, and save a machine learning model to your filesystem or a cloud storage provider.

data-science keras machine-learning mlops modelstore python-library pytorch s3-storage scikit-learn tensorflow transformer

Last synced: 02 Aug 2024

https://github.com/ReshotAI/gaussian-painters

Gaussian Painters using 3D Gaussian Splatting

3d ai art gaussian-splatting machine-learning nerf painting python pytorch

Last synced: 01 Aug 2024

https://github.com/NeuromorphicProcessorProject/snn_toolbox

Toolbox for converting analog to spiking neural networks (ANN to SNN), and running them in a spiking neuron simulator.

brian brian2 caffe deep-learning deep-neural-networks keras lasagne loihi nest pynn pytorch spiking-neural-networks spinnaker tensorflow

Last synced: 02 Aug 2024

https://github.com/BlackHC/toma

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory

data-science gpu machine-learning python pytorch

Last synced: 03 Aug 2024

https://github.com/kefirski/pytorch_RVAE

Recurrent Variational Autoencoder that generates sequential data implemented with pytorch

deep-learning nlp python pytorch vae

Last synced: 01 Aug 2024

https://github.com/kakaoenterprise/JORLDY

Repository for Open Source Reinforcement Learning Framework JORLDY

framework gym mlagents pytorch reinforcement-learning

Last synced: 08 Aug 2024

https://github.com/dptech-corp/Uni-Fold

An open-source platform for developing protein models beyond AlphaFold.

alphafold alphafold-multimer deep-learning protein-structure pytorch

Last synced: 02 Aug 2024

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

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

https://github.com/Haiyang-W/DSVT

[CVPR2023] Official Implementation of "DSVT: Dynamic Sparse Voxel Transformer with Rotated Sets"

3d-object-detection backbone cvpr2023 point-cloud pytorch transformer

Last synced: 31 Jul 2024

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

yolov3 network slimming剪枝的一种实现

network pruning pytorch slimming yolo

Last synced: 02 Aug 2024

https://github.com/prigoyal/pytorch_memonger

Experimental ground for optimizing memory of pytorch models

memory-management pytorch

Last synced: 01 Aug 2024

https://github.com/emidan19/deep-tempest

Restoration for TEMPEST images using deep-learning

deep-learning gnu-radio image-restoration pytorch sdr tempest

Last synced: 08 Aug 2024

https://github.com/square/pysurvival

Open source package for Survival Analysis modeling

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

Last synced: 02 Aug 2024

https://github.com/ymcui/PERT

PERT: Pre-training BERT with Permuted Language Model

bert nlp plm pre-trained-model pytorch tensorflow transformers

Last synced: 03 Aug 2024

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

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: 31 Jul 2024

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

Evolution Strategies in PyTorch

machinelearning python pytorch

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

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

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: 31 Jul 2024

https://github.com/Pointcept/PointTransformerV2

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

3d-vision point-cloud pytorch

Last synced: 31 Jul 2024

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

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

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

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

pytorch pytorch-nlp pytorch-tutorial

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

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

PyTorch C++ inference with LibTorch

cpp demo imagenet inference libtorch opencv pytorch pytorch-cpp

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

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: 31 Jul 2024

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

https://github.com/iMoonLab/MeshNet

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

3d-shapes deep-learning mesh pytorch

Last synced: 31 Jul 2024

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

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

augmentation pytorch pytorch-lightning tensorflow torchbearer

Last synced: 02 Aug 2024

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

https://github.com/wohlert/generative-query-network-pytorch

Generative Query Network (GQN) in PyTorch as described in "Neural Scene Representation and Rendering"

deepmind generative-models gqn pytorch

Last synced: 07 Aug 2024

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

https://github.com/Sierkinhane/mtcnn-pytorch

A face detection algorithm

computer-vision facedetector pytorch

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

https://github.com/damitkwr/ESRNN-GPU

PyTorch GPU implementation of the ES-RNN model for time series forecasting

deep-forecasting deep-learning es-rnn forecasting pytorch time-series-forecasting

Last synced: 01 Aug 2024

https://github.com/JUGGHM/PENet_ICRA2021

ICRA 2021 "Towards Precise and Efficient Image Guided Depth Completion"

depth-completion pytorch

Last synced: 31 Jul 2024

https://github.com/yqyao/FCOS_PLUS

Some improvements (center sample) about FCOS (FCOS: Fully Convolutional One-Stage Object Detection).

anchor-free computer-vision fcos object-detection one-stage pytorch

Last synced: 05 Aug 2024

https://github.com/rlgraph/rlgraph

RLgraph: Modular computation graphs for deep reinforcement learning

deep-learning deep-reinforcement-learning dqn machine-learning neural-networks ppo pytorch reinforcement-learning tensorflow

Last synced: 01 Aug 2024

https://github.com/chenxi116/PNASNet.pytorch

PyTorch implementation of PNASNet-5 on ImageNet

automl deep-learning imagenet neural-architecture-search pytorch

Last synced: 31 Jul 2024

https://github.com/gasteigerjo/ppnp

PPNP & APPNP models from "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019)

deep-learning gcn gnn graph-algorithms graph-classification graph-neural-networks machine-learning pagerank pytorch tensorflow

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

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

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

https://github.com/jzi040941/PercepNet

Unofficial implementation of PercepNet: A Perceptually-Motivated Approach for Low-Complexity, Real-Time Enhancement of Fullband Speech

pytorch speech-enhancement

Last synced: 03 Aug 2024

https://github.com/IBM/aihwkit

IBM Analog Hardware Acceleration Kit

ai analog-devices cuda neural-networks pytorch

Last synced: 03 Aug 2024