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.

GitHub: https://github.com/topics/pytorch
Wikipedia: https://en.wikipedia.org/wiki/PyTorch
Repo: https://github.com/pytorch/pytorch
Created by: Facebook's AI Research lab (FAIR)
Released: September 2016
Related Topics: python, pytorch-tutorial,
Last updated: 2025-02-12 00:24:38 UTC
JSON Representation

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

https://github.com/gentaiscool/end2end-asr-pytorch

End-to-End Automatic Speech Recognition on PyTorch

asr end-to-end pytorch speech speech-recognition transformer

Last synced: 07 Feb 2025

https://github.com/AssemblyAI-Community/MinImagen

MinImagen: A minimal implementation of the Imagen text-to-image model

deep-learning diffusion-models imagen pytorch super-resolution text-to-image

Last synced: 25 Nov 2024

https://github.com/nikhilbarhate99/Hierarchical-Actor-Critic-HAC-PyTorch

PyTorch implementation of Hierarchical Actor Critic (HAC) for OpenAI gym environments

actor-critic deep-reinforcement-learning gym-environment gym-environments hierarchical-reinforcement-learning openai-gym pytorch pytorch-implementation pytorch-rl reinforcement-learning reinforcement-learning-algorithms

Last synced: 16 Nov 2024

https://github.com/gmongaras/Diffusion_models_from_scratch

Creating a diffusion model from scratch in PyTorch to learn exactly how they work.

classifier-free-guidance ddim ddpm diffusion-models imagenet pytorch

Last synced: 30 Oct 2024

https://github.com/Luodian/Generalizable-Mixture-of-Experts

GMoE could be the next backbone model for many kinds of generalization task.

deep-learning domain-generalization mixture-of-experts pytorch pytorch-implementation

Last synced: 15 Nov 2024

https://github.com/jaxony/shufflenet

ShuffleNet in PyTorch. Based on https://arxiv.org/abs/1707.01083

artificial-intelligence convolution deep-learning neural-network pytorch

Last synced: 07 Feb 2025

https://github.com/pengboxiangshang/multigraph_transformer

IEEE TNNLS 2021, transformer, multi-graph transformer, graph, graph classification, sketch recognition, sketch classification, free-hand sketch, official code of the paper "Multi-Graph Transformer for Free-Hand Sketch Recognition"

graph multi-graph-transformer pytorch pytorch-implementation sketch sketch-recognition sparse-graphs transformer transformer-architecture

Last synced: 07 Feb 2025

https://github.com/feedly/transfer-nlp

NLP library designed for reproducible experimentation management

framework language-model natural-language-understanding nlp playground pytorch transfer-learning

Last synced: 08 Feb 2025

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

[CVPR'2022] SAM-DETR & SAM-DETR++: Official PyTorch Implementation

computer-vision cvpr cvpr2022 deep-learning detection detr machine-learning object-detection pytorch transformer vision vision-transformer

Last synced: 28 Oct 2024

https://github.com/lucastabelini/PolyLaneNet

Code for the paper entitled "PolyLaneNet: Lane Estimation via Deep Polynomial Regression" (ICPR 2020)

deep-learning lane-detection pytorch

Last synced: 28 Oct 2024

https://github.com/xptree/deepinf

DeepInf: Social Influence Prediction with Deep Learning

graph-attention-networks graph-convolutional-networks pytorch social-network

Last synced: 07 Feb 2025

https://github.com/Bluefog-Lib/bluefog

Distributed and decentralized training framework for PyTorch over graph

asynchronous decentralized deeplearning distributed-computing machine-learning mpi nccl one-sided pytorch

Last synced: 27 Nov 2024

https://github.com/wonjongg/StyleCariGAN

[SIGGRAPH 2021] Official implementation of "StyleCariGAN: Caricature Generation via StyleGAN Feature Map Modulation"

deep-learning pytorch

Last synced: 27 Nov 2024

https://github.com/tky823/dnn-based_source_separation

A PyTorch implementation of DNN-based source separation.

audio-separation conv-tasnet pytorch source-separation speech-separation tasnet

Last synced: 06 Feb 2025

https://github.com/maudzung/rtm3d

Unofficial PyTorch implementation of "RTM3D: Real-time Monocular 3D Detection from Object Keypoints for Autonomous Driving" (ECCV 2020)

3d-object-detection autonomous-driving autonomous-vehicles centernet kitti-dataset monocular-images pytorch pytorch-implementation real-time rtm3d self-driving-car

Last synced: 10 Feb 2025

https://github.com/nateraw/huggingpics

🤗🖼️ HuggingPics: Fine-tune Vision Transformers for anything using images found on the web.

huggingface machine-learning pytorch vision-transformers

Last synced: 09 Feb 2025

https://github.com/fcakyon/yolov5-pip

Packaged version of ultralytics/yolov5 + many extra features

aws cli coco computer-vision deep-learning machine-learning neptune neptune-ai object-detection pip pypi python pytorch s3 ultralytics yolo yolov3 yolov4 yolov5

Last synced: 07 Feb 2025

https://github.com/chandlerbang/pro-gnn

Implementation of the KDD 2020 paper "Graph Structure Learning for Robust Graph Neural Networks"

adversarial-attacks attack-defense deep-learning defense graph-mining graph-neural-networks graph-structure-recovery machine-learning pytorch semi-supervised-learning

Last synced: 07 Feb 2025

https://github.com/cadene/tensorflow-model-zoo.torch

InceptionV3, InceptionV4, Inception-Resnet pretrained models for Torch7 and PyTorch

inception pretrained-models pytorch tensorflow torch

Last synced: 06 Feb 2025

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: 10 Feb 2025

https://github.com/awslabs/adatune

Gradient based Hyperparameter Tuning library in PyTorch

automl deep-learning hyperparameter-tuning learning-rate-scheduling machine-learning neural-networks pytorch

Last synced: 16 Nov 2024

https://github.com/jaxony/ShuffleNet

ShuffleNet in PyTorch. Based on https://arxiv.org/abs/1707.01083

artificial-intelligence convolution deep-learning neural-network pytorch

Last synced: 31 Oct 2024

https://github.com/leaderj1001/MobileNetV3-Pytorch

Implementing Searching for MobileNetV3 paper using Pytorch

cifar-10 cifar-100 imagenet mobilenetv3 pytorch searching-for-mobilenetv3

Last synced: 31 Oct 2024

https://github.com/ksw0306/ClariNet

A Pytorch Implementation of ClariNet

clarinet parallel-wavenet pytorch wavenet

Last synced: 04 Nov 2024

https://github.com/lucidrains/routing-transformer

Fully featured implementation of Routing Transformer

artificial-intelligence attention-mechanism deep-learning pytorch transformer

Last synced: 09 Feb 2025

https://github.com/torchspatiotemporal/tsl

tsl: a PyTorch library for processing spatiotemporal data.

deep-learning gnn graph-neural-networks pytorch spatio-temporal spatio-temporal-analysis spatio-temporal-data spatio-temporal-graph spatio-temporal-prediction spatiotemporal spatiotemporal-data spatiotemporal-data-analysis spatiotemporal-forecasting temporal-data temporal-graphs

Last synced: 08 Feb 2025

https://github.com/sajjjadayobi/FaceLib

Face Analysis: Detection, Age Gender Estimation & Recognition

age-gender deep-learning face face-detection face-recognition facial-expression-recognition gender-estimation pytorch recognition retinaface

Last synced: 20 Nov 2024

https://github.com/guochengqian/tenet

[ICCP'22] Rethinking Learning-based Demosaicing, Denoising, and Super-Resolution Pipeline

demosaicing denoising pytorch super-resolution

Last synced: 06 Feb 2025

https://github.com/idea-research/humansd

[ICCV 2023] The official implementation of paper "HumanSD: A Native Skeleton-Guided Diffusion Model for Human Image Generation"

conditional-image-generation deep-learning iccv iccv2023 image-generation pytorch

Last synced: 07 Feb 2025

https://github.com/qitianwu/DIFFormer

The official implementation for ICLR23 spotlight paper "DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusion"

attention diffusion diffusion-equation geometric-deep-learning graph-neural-networks graph-transformer iclr2023 image-classification large-graph node-classification pytorch pytorch-geometric pytorch-geometric-temporal spatial-temporal-forecasting text-classification transformer

Last synced: 30 Oct 2024

https://github.com/gabrielilharco/snap-n-eat

Food detection and recommendation with deep learning

deep-learning deep-neural-networks fastai flask food food-classification food-recommendation knn nodejs pytorch resnext

Last synced: 06 Feb 2025

https://github.com/scikit-hep/pyhf

pure-Python HistFactory implementation with tensors and autodiff

asymptotic-formulas closember cls frequentist-statistics hep hep-ex high-energy-physics histfactory jax numpy python pytorch scientific-computations scikit-hep scipy statistical-inference statistics tensorflow

Last synced: 07 Feb 2025

https://github.com/suyukun666/UFO

Official PyTorch implementation of the “A Unified Transformer Framework for Co-Segmentation, Co-Saliency Detection and Video Salient Object Detection”. (TMM2023)

co-saliency-detection co-segmentation deep-learning inpainting matching pytorch video video-salient-object-detection

Last synced: 11 Nov 2024

https://github.com/lukaszahradnik/pyneuralogic

PyNeuraLogic lets you use Python to create Differentiable Logic Programs

deep-learning differentiable-programming geometric-deep-learning graph-neural-networks logic-programming machine-learning python pytorch relational-learning

Last synced: 08 Feb 2025

https://github.com/fabio-sim/depth-anything-onnx

ONNX-compatible Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data

deep-learning depth depth-anything depth-estimation machine-learning monocular-depth-estimation onnx onnxruntime pytorch

Last synced: 08 Feb 2025

https://github.com/Media-Smart/volksdep

volksdep is an open-source toolbox for deploying and accelerating PyTorch, ONNX and TensorFlow models with TensorRT.

acceleration deploy inference jetson-nano jetson-tx2 jetson-xavier keras onnx python pytorch tensorflow tensorrt

Last synced: 14 Nov 2024

https://github.com/dena/handyrl

HandyRL is a handy and simple framework based on Python and PyTorch for distributed reinforcement learning that is applicable to your own environments.

deep-learning distributed-training games machine-learning policy-gradient pytorch reinforcement-learning

Last synced: 05 Feb 2025

https://github.com/bytedance/flux

A fast communication-overlapping library for tensor parallelism on GPUs.

cuda cutlass gpu pytorch

Last synced: 08 Feb 2025

https://github.com/sgrvinod/a-pytorch-tutorial-to-transformers

Attention Is All You Need | a PyTorch Tutorial to Transformers

attention-is-all-you-need pytorch pytorch-tutorial transformer transformer-architecture transformer-tutorial

Last synced: 06 Feb 2025

https://github.com/dgkim5360/the-elements-of-statistical-learning-notebooks

Jupyter notebooks for summarizing and reproducing the textbook "The Elements of Statistical Learning" 2/E by Hastie, Tibshirani, and Friedman

jupyter-notebooks machine-learning matplotlib neural-network python3 pytorch scipy statistical-learning textbook

Last synced: 07 Feb 2025

https://github.com/loeweX/Greedy_InfoMax

Code for the paper: Putting An End to End-to-End: Gradient-Isolated Learning of Representations

deep-learning local-learning pytorch

Last synced: 12 Nov 2024

https://github.com/outcastofmusic/quick-nlp

Pytorch NLP library based on FastAI

fastai nlp-library pytorch seq2seq

Last synced: 14 Nov 2024

https://github.com/ChandlerBang/Pro-GNN

Implementation of the KDD 2020 paper "Graph Structure Learning for Robust Graph Neural Networks"

adversarial-attacks attack-defense deep-learning defense graph-mining graph-neural-networks graph-structure-recovery machine-learning pytorch semi-supervised-learning

Last synced: 28 Nov 2024

https://github.com/xyupeng/ContrastiveCrop

[CVPR 2022 Oral] Crafting Better Contrastive Views for Siamese Representation Learning

contrastive-learning pytorch self-supervised-learning

Last synced: 15 Nov 2024

https://github.com/kwea123/casmvsnet_pl

Cascade Cost Volume for High-Resolution Multi-View Stereo and Stereo Matching using pytorch-lightning

3d-reconstruction blendedmvs cascade-cost-volume depth-prediction dtu-dataset multi-view-stereo mvsnet pytorch pytorch-lightning tanks-and-temples

Last synced: 06 Feb 2025

https://github.com/mantasu/cs231n

Shortest solutions for CS231n 2021-2024

computer-vision convolutional-neural-networks cs231n deep-learning pytorch tensorflow visual-recognition

Last synced: 05 Feb 2025

https://github.com/rballester/tntorch

Tensor Network Learning with PyTorch

cp-decomposition data-science learning pytorch tensor-decomposition tensor-networks tensor-train tensors tucker-decomposition

Last synced: 15 Nov 2024

https://github.com/junyaohu/common_metrics_on_video_quality

You can easily calculate FVD, PSNR, SSIM, LPIPS for evaluating the quality of generated or predicted videos.

fvd metrics psnr pytorch ssim video video-generation video-prediction

Last synced: 06 Feb 2025

https://github.com/fregu856/3DOD_thesis

3D Object Detection for Autonomous Driving in PyTorch, trained on the KITTI dataset.

3d-deep-learning 3d-detection autonomous-driving computer-vision deep-learning machine-learning object-detection pytorch

Last synced: 27 Nov 2024

https://github.com/aimagelab/show-control-and-tell

Show, Control and Tell: A Framework for Generating Controllable and Grounded Captions. CVPR 2019

caption-generation captioning-images cvpr2019 image-captioning pytorch visual-semantic

Last synced: 06 Feb 2025

https://github.com/pmj110119/YOLOX_deepsort_tracker

using yolox+deepsort for object-tracking

deep-sort deepsort object-detection object-tracker pytorch tracker yolo yolo-x yolov3 yolov5 yolox

Last synced: 09 Nov 2024

https://github.com/shoufachen/cyclemlp

[ICLR'22 Oral] Implementation of "CycleMLP: A MLP-like Architecture for Dense Prediction"

backbone-models mlp pytorch

Last synced: 24 Jan 2025

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

NU-Wave 2: A General Neural Audio Upsampling Model for Various Sampling Rates @ INTERSPEECH 2022

deep-learning neural-audio-upsampling pytorch super-resolution upsampling

Last synced: 07 Feb 2025

https://github.com/numenta/nupic.torch

Numenta Platform for Intelligent Computing PyTorch libraries

htm numenta-platform pytorch

Last synced: 10 Feb 2025

https://github.com/merantix-momentum/squirrel-core

A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way :chestnut:

ai cloud-computing collaboration computer-vision cv data-ingestion data-mesh data-science dataops datasets deep-learning distributed internal machine-learning ml natural-language-processing nlp python pytorch tensorflow

Last synced: 02 Nov 2024

https://github.com/sunzeyeah/RLHF

Implementation of Chinese ChatGPT

chatgpt deep-learning deepspeed glm nlp pangu pytorch

Last synced: 31 Oct 2024

https://github.com/DeNA/HandyRL

HandyRL is a handy and simple framework based on Python and PyTorch for distributed reinforcement learning that is applicable to your own environments.

deep-learning distributed-training games machine-learning policy-gradient pytorch reinforcement-learning

Last synced: 04 Nov 2024

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

NU-Wave: A Diffusion Probabilistic Model for Neural Audio Upsampling @ INTERSPEECH 2021

deep-generative-model deep-learning neural-audio-upsampling pytorch super-resolution upsampling

Last synced: 09 Feb 2025

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

[ICLR2024] Official repo for paper "PnP Inversion: Boosting Diffusion-based Editing with 3 Lines of Code"

deep-learning diffusion-inversion diffusion-model diffusion-models iclr2024 image-editing image-generation paper-implementation paper-implementations pytorch

Last synced: 07 Feb 2025

https://github.com/extreme-bert/extreme-bert

ExtremeBERT is a toolkit that accelerates the pretraining of customized language models on customized datasets, described in the paper “ExtremeBERT: A Toolkit for Accelerating Pretraining of Customized BERT”.

bert deep-learning language-model language-models machine-learning natural-language-processing nlp python pytorch transformer

Last synced: 16 Nov 2024

https://github.com/pooya-mohammadi/yolov5-gradcam

Visualizing Yolov5's layers using GradCam

convolutional-neural-networks deep-learning deeputils gradcam pytorch visualizing yolov5

Last synced: 09 Nov 2024

https://github.com/DocF/multispectral-object-detection

Multispectral Object Detection with Yolov5 and Transformer

deep-learning object-detection pytorch transformer yolov5

Last synced: 09 Nov 2024

https://github.com/showlab/all-in-one

[CVPR2023] All in One: Exploring Unified Video-Language Pre-training

codebase pre-training pytorch video-language

Last synced: 07 Feb 2025

https://github.com/krishnap25/mauve

Package to compute Mauve, a similarity score between neural text and human text. Install with `pip install mauve-text`.

deep-learning huggingface-transformers nlp pytorch text-generation

Last synced: 08 Feb 2025

https://github.com/nvidia/bionemo-framework

BioNeMo Framework: For building and adapting AI models in drug discovery at scale

drug-discovery gpu machine-learning pytorch

Last synced: 08 Feb 2025

https://github.com/jacobgil/pytorch-tensor-decompositions

PyTorch implementation of [1412.6553] and [1511.06530] tensor decomposition methods for convolutional layers.

deep-learning pytorch tensorly

Last synced: 09 Feb 2025

https://github.com/awslabs/renate

Library for automatic retraining and continual learning

aws continual-learning hyperparameter-tuning hyperparameters-optimization machine-learning machine-learning-algorithms neural neural-network pytorch pytorch-lightning sagemaker

Last synced: 08 Feb 2025

https://github.com/tugstugi/pytorch-saltnet

Kaggle | 9th place single model solution for TGS Salt Identification Challenge

cnn convolutional-neural-networks deep-learning kaggle-competition pytorch segmentation seismic-imaging unet unet-image-segmentation unet-pytorch

Last synced: 23 Jan 2025

https://github.com/jingweiz/pytorch-dnc

Neural Turing Machine (NTM) & Differentiable Neural Computer (DNC) with pytorch & visdom

deep-learning dnc external-memory ntm pytorch visdom

Last synced: 21 Nov 2024

https://github.com/iArunava/ENet-Real-Time-Semantic-Segmentation

ENet - A Neural Net Architecture for real time Semantic Segmentation

artificial-intelligence deep-learning enet paper python3 pytorch

Last synced: 27 Nov 2024

https://github.com/kkanshul/finegan

FineGAN: Unsupervised Hierarchical Disentanglement for Fine-grained Object Generation and Discovery

computer-vision deep-learning disentangled-representations fine-grained gans generative-adversarial-network image-generation image-manipulation pytorch

Last synced: 15 Nov 2024

https://github.com/delijati/pytorch-siamese

Siamese Network implementation using Pytorch

contrastive pytorch siamese

Last synced: 06 Feb 2025

https://github.com/Lin-Yijie/Graph-Matching-Networks

PyTorch implementation of Graph Matching Networks, e.g., Graph Matching with Bi-level Noisy Correspondence (COMMON, ICCV 2023), Graph Matching Networks for Learning the Similarity of Graph Structured Objects (GMN, ICML 2019).

deep-learning gnn graph-neural-networks pytorch

Last synced: 04 Nov 2024

https://github.com/millionintegrals/vel

Velocity in deep-learning research

convolutional-neural-networks deep-learning python pytorch reinforcement-learning

Last synced: 10 Feb 2025

https://github.com/MillionIntegrals/vel

Velocity in deep-learning research

convolutional-neural-networks deep-learning python pytorch reinforcement-learning

Last synced: 02 Nov 2024

https://github.com/szagoruyko/pyinn

CuPy fused PyTorch neural networks ops

cupy pytorch

Last synced: 24 Jan 2025

https://github.com/deephyper/deephyper

DeepHyper: Scalable Asynchronous Neural Architecture and Hyperparameter Search for Deep Neural Networks

automl deep-learning hpc hyperparameter-optimization keras machine-learning mpi multi-fidelity neural-architecture-search python pytorch raylib scalability tensorflow uncertainty-quantification

Last synced: 07 Feb 2025

https://github.com/jaxony/unet-pytorch

U-Net implementation for PyTorch based on https://arxiv.org/abs/1505.04597

deep-learning neural-network python pytorch semantic-segmentation

Last synced: 24 Jan 2025

https://github.com/sdsubhajitdas/brain-tumor-segmentation

Brain Tumor Segmentation done using U-Net Architecture.

brain-tumor-segmentation deep-learning image-segmentation pytorch tumor-segmentation unet unet-image-segmentation unet-pytorch

Last synced: 06 Feb 2025

https://github.com/prodramp/DeepWorks

A collection of Deep Learning projects and resources

artificial-intelligence deep-learning generative-models llm machine-learning pytorch tensorflow

Last synced: 29 Oct 2024

https://github.com/dyhan0920/PyramidNet-PyTorch

A PyTorch implementation for PyramidNets (Deep Pyramidal Residual Networks, https://arxiv.org/abs/1610.02915)

cifar-10 cifar-100 imagenet pyramidnet pytorch residual-networks resnet

Last synced: 31 Oct 2024

https://github.com/rentainhe/pytorch-distributed-training

Simple tutorials on Pytorch DDP training

apex cuda ddp-training deep-learning pytorch

Last synced: 07 Feb 2025

https://github.com/AnjieCheng/MnasNet-PyTorch

A PyTorch implementation of Mnasnet: MnasNet: Platform-Aware Neural Architecture Search for Mobile.

architecture-search mnasnet mobilenetv2 pytorch

Last synced: 31 Oct 2024

https://github.com/toluclassics/candle-tutorial

Tutorial for Porting PyTorch Transformer Models to Candle (Rust)

candle pytorch rust rust-lang

Last synced: 04 Feb 2025

https://github.com/benedekrozemberczki/sgcn

A PyTorch implementation of "Signed Graph Convolutional Network" (ICDM 2018).

deep-learning deepwalk gcn gpt2 gpt3 graph-attention graph-convolution graph-embedding graph-neural-networks graphsage machine-learning network-embedding neural-network node-classification node2vec pytorch pytorch-geometric sgcn signed-network transformer

Last synced: 07 Feb 2025

https://github.com/WenjieDu/SAITS

The official PyTorch implementation of the paper "SAITS: Self-Attention-based Imputation for Time Series". A fast and state-of-the-art (SOTA) deep-learning neural network model for efficient time-series imputation (impute multivariate incomplete time series containing NaN missing data/values with machine learning). https://arxiv.org/abs/2202.08516

attention attention-mechanism deep-learning imputation imputation-model impute incomplete-data incomplete-time-series interpolation irregular-sampling machine-learning missing-values partially-observed partially-observed-data partially-observed-time-series pytorch self-attention time-series time-series-imputation transformer

Last synced: 02 Nov 2024

https://github.com/neulab/knn-transformers

PyTorch + HuggingFace code for RetoMaton: "Neuro-Symbolic Language Modeling with Automaton-augmented Retrieval" (ICML 2022), including an implementation of kNN-LM and kNN-MT

huggingface knn knn-lm knn-mt knn-transformers knnlm knnmt language language-models machine models nearest nearest-neighbor neighbor neuro-symbolic pytorch retomaton transformers translation

Last synced: 23 Jan 2025

https://github.com/benedekrozemberczki/gam

A PyTorch implementation of "Graph Classification Using Structural Attention" (KDD 2018).

attention deep-reinforcement-learning deepwalk gcn graph-attention graph-classification graph-convolution graph2vec graphsage machine-learning neural-networks node2vec pytorch reinforcement-agent reinforcement-learning sklearn structural-attention transfer-learning

Last synced: 07 Feb 2025

https://github.com/xpixelgroup/ranksrgan

ICCV 2019 (oral) RankSRGAN: Generative Adversarial Networks with Ranker for Image Super-Resolution. PyTorch implementation

gan generative-adversarial-network iccv2019 learning-to-rank low-level-vision pytorch super-resolution

Last synced: 23 Jan 2025

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

Unofficial PyTorch Implementation of UnivNet Vocoder (https://arxiv.org/abs/2106.07889)

deep-learning gan pytorch speech-synthesis text-to-speech tts vocoder

Last synced: 07 Feb 2025

https://github.com/ducha-aiki/affnet

Code and weights for local feature affine shape estimation paper "Repeatability Is Not Enough: Learning Discriminative Affine Regions via Discriminability"

affine-shape-estimator computer-vision convolutional-networks convolutional-neural-networks deep-learning hessian image-matching image-retrieval local-features pytorch