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/ZhixiuYe/Intra-Bag-and-Inter-Bag-Attentions

Code for NAACL 2019 paper: Distant Supervision Relation Extraction with Intra-Bag and Inter-Bag Attentions

deeplearning distant-supervision nlp pytorch relation-extraction

Last synced: 01 Nov 2024

https://github.com/clovaai/focusseq2seq

[EMNLP 2019] Mixture Content Selection for Diverse Sequence Generation (Question Generation / Abstractive Summarization)

emnlp2019 generation nlp pytorch question-generation summarization

Last synced: 12 Nov 2024

https://github.com/yingkunwu/R-YOLOv4

This is a PyTorch-based R-YOLOv4 implementation which combines YOLOv4 model and loss function from R3Det for arbitrary oriented object detection.

oriented-object-detection pytorch yolov4

Last synced: 09 Nov 2024

https://github.com/jiaweihe1996/GMTracker

Official PyTorch implementation of "Learnable Graph Matching: Incorporating Graph Partitioning with Deep Feature Learning for Multiple Object Tracking" (CVPR 2021).

deep-learning pytorch tracking-by-detection

Last synced: 13 Nov 2024

https://github.com/pytorch/extension-script

Example repository for custom C++/CUDA operators for TorchScript

cpp cuda pytorch

Last synced: 26 Sep 2024

https://github.com/shibing624/nerpy

🌈 NERpy: Implementation of Named Entity Recognition using Python. 命名实体识别工具,支持BertSoftmax、BertSpan等模型,开箱即用。

bert bert-softmax bert-span named-entity-recognition ner nlp pytorch transformers

Last synced: 14 Nov 2024

https://github.com/airaria/captcharecognition

End-to-end variable length Captcha recognition using CNN+RNN+Attention/CTC (pytorch implementation). 端到端的不定长验证码识别

attention cnn pytorch rnn

Last synced: 11 Nov 2024

https://github.com/haoxiangsnr/IRM-based-Speech-Enhancement-using-LSTM

Ideal Ratio Mask (IRM) Estimation based Speech Enhancement using LSTM

deep-neural-networks ideal-ratio-mask lstm pytorch speech-enhancement

Last synced: 02 Nov 2024

https://github.com/4uiiurz1/pytorch-res2net

PyTorch implementation of Res2Net

deep-learning pytorch res2net

Last synced: 31 Oct 2024

https://github.com/Saro00/DGN

Implementation of Directional Graph Networks in PyTorch and DGL

dgl graph-machine-learning graph-neural-networks pytorch spectral-methods

Last synced: 08 Aug 2024

https://github.com/narumiruna/efficientnet-pytorch

A PyTorch implementation of "EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks".

cifar10 efficientnet imagenet mnist pytorch

Last synced: 13 Nov 2024

https://github.com/aphp/edsnlp

Modular, fast NLP framework, compatible with Pytorch and spaCy, offering tailored support for French clinical notes.

clinical-data-warehouse deep-learning fast french medical multi-task nlp pytorch rule-based spacy text-mining

Last synced: 14 Oct 2024

https://github.com/zhanghengdev/MutualGuide

Localize to Classify and Classify to Localize: Mutual Guidance in Object Detection

deep-learning knowledge-distillation object-detection pytorch tensorrt

Last synced: 08 Nov 2024

https://github.com/Vermeille/Torchelie

Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.

gan loss perceptual pytorch torch utils vqvae

Last synced: 12 Nov 2024

https://github.com/bshall/tacotron

A PyTorch implementation of Location-Relative Attention Mechanisms For Robust Long-Form Speech Synthesis

attention-mechanism pytorch speech-synthesis tacotron text-to-speech tts

Last synced: 11 Oct 2024

https://github.com/aa-samad/conv_snn

Code for "Convolutional spiking neural networks (SNN) for spatio-temporal feature extraction" paper

cifar10 cnn convolutional-neural-networks deep-learning neuromorphic pytorch snn spatio-temporal-analysis

Last synced: 13 Nov 2024

https://github.com/JD-AI-Research-Silicon-Valley/SACN

End-to-end Structure-Aware Convolutional Networks for Knowledge Base Completion

graph-convolutional-networks knowledge-graph knowledge-graph-completion pytorch sacn

Last synced: 08 Aug 2024

https://github.com/lzhbrian/metrics

IS, FID score Pytorch and TF implementation, TF implementation is a wrapper of the official ones.

api commandline fid frechet-distance frechet-inception-distance gan inception inception-score is metrics pytorch tensorflow

Last synced: 07 Nov 2024

https://github.com/Qrange-group/SUR-adapter

ACM MM'23 (oral), SUR-adapter for pre-trained diffusion models can acquire the powerful semantic understanding and reasoning capabilities from large language models to build a high-quality textual semantic representation for text-to-image generation.

adapter diffusion-models image-generation knowledge-distillation large-language-models pytorch

Last synced: 31 Oct 2024

https://github.com/ucbrise/graphtrans

Representing Long-Range Context for Graph Neural Networks with Global Attention

deep-learning graph-neural-networks pytorch transformer

Last synced: 07 Nov 2024

https://github.com/dahyun-kang/renet

[ICCV'21] Official PyTorch implementation of Relational Embedding for Few-Shot Classification

computer-vision deep-learning few-shot-classifcation few-shot-learning iccv2021 neural-networks pytorch

Last synced: 15 Nov 2024

https://github.com/nju-websoft/muKG

μKG: A Library for Multi-source Knowledge Graph Embeddings and Applications, ISWC 2022

deep-learning entity-alignment entity-typing link-prediction multi-source-knowledge-graphs pytorch representation-learning

Last synced: 15 Nov 2024

https://github.com/pooya-mohammadi/deep_utils

An open-source toolkit which is full of handy functions, including the most used models and utilities for deep-learning practitioners!

augmentation coco computer-vision cutmix deep-learning face-detection face-recognition machine-learning modelcheckpoint nlp object-detection python pytorch senet tensorflow utils vggface2 yolov5

Last synced: 09 Nov 2024

https://github.com/juliagusak/dataloaders

Pytorch and TensorFlow data loaders for several audio datasets

audio-processing dataloader dataset esc gtzan librispeech nsynth pytorch tfrecords

Last synced: 09 Nov 2024

https://github.com/kuangliu/pytorch-ssd

Single Shot MultiBox Detector in PyTorch [deprecated]

pytorch ssd

Last synced: 10 Nov 2024

https://github.com/funcwj/deep-clustering

deep clustering method for single-channel speech separation

pytorch speech-separation

Last synced: 02 Nov 2024

https://github.com/chendrag/mujoco-benchmark

Provide full reinforcement learning benchmark on mujoco environments, including ddpg, sac, td3, pg, a2c, ppo, library

baseline benchmark ddpg drl mujoco performance ppo pytorch results rl sac tianshou

Last synced: 27 Oct 2024

https://github.com/shoprunner/collie

A library for preparing, training, and evaluating scalable deep learning hybrid recommender systems using PyTorch.

datascience deep-learning hybrid implicit learning-to-rank python pytorch recommendation-system recommender-system

Last synced: 14 Nov 2024

https://github.com/S-aiueo32/srntt-pytorch

A PyTorch implementation of SRNTT, which is a novel RefSR method.

paper-implementations pytorch reference-based refsr srntt super-resolution

Last synced: 11 Nov 2024

https://github.com/cardwing/codes-for-intra-kd

Inter-Region Affinity Distillation for Road Marking Segmentation (CVPR 2020)

cnn deep-learning lane-detection pytorch road-marking-segmentation

Last synced: 30 Oct 2024

https://github.com/stepelu/ptstat

Probabilistic Programming and Statistical Inference in PyTorch

probability pytorch statistics

Last synced: 14 Nov 2024

https://github.com/rotten-work/vits-mandarin-windows

VITS for Mandarin. Support Windows and Linux, low-end and high-end hardwares

chinese linux mandarin pytorch vits windows-desktop

Last synced: 03 Aug 2024

https://github.com/yihongXU/TransCenter

This is the official implementation of TransCenter (TPAMI). The code and pretrained models are now available here: https://gitlab.inria.fr/yixu/TransCenter_official.

computer-vision deep-learning multiple-object-tracking pytorch transformers

Last synced: 13 Nov 2024

https://github.com/khurramjaved96/incremental-learning

Pytorch implementation of ACCV18 paper "Revisiting Distillation and Incremental Classifier Learning."

convolutional-neural-networks distillation incremental-learning machine-learning paper-implementations pytorch

Last synced: 15 Nov 2024

https://github.com/vita-group/gan-slimming

[ECCV 2020] "All-in-One GAN Compression by Unified Optimization" by Haotao Wang, Shupeng Gui, Haichuan Yang, Ji Liu, and Zhangyang Wang

compression cyclegan gan pytorch

Last synced: 16 Nov 2024

https://github.com/xiaosu-zhu/mcquic

Repository of CVPR'22 paper "Unified Multivariate Gaussian Mixture for Efficient Neural Image Compression"

computer-vision cvpr2022 image-compression image-processing pytorch

Last synced: 02 Oct 2024

https://github.com/laddie132/Match-LSTM

A PyTorch implemention of Match-LSTM, R-NET and M-Reader for Machine Reading Comprehension

machine-comprehension match-lstm mrc pytorch r-net squad

Last synced: 07 Aug 2024

https://github.com/BarqueroGerman/BeLFusion

[ICCV2023] Official PyTorch Implementation of "BeLFusion: Latent Diffusion for Behavior-Driven Human Motion Prediction". ICCV 2023

belfusion ddim ddpm deep-learning diffusion generative-models human-motion-generation human-motion-prediction iccv2023 latent-diffusion ldm motion-forecasting motion-prediction pose-forecasting pose-prediction pytorch

Last synced: 10 Nov 2024

https://barquerogerman.github.io/BeLFusion/

[ICCV2023] Official PyTorch Implementation of "BeLFusion: Latent Diffusion for Behavior-Driven Human Motion Prediction". ICCV 2023

belfusion ddim ddpm deep-learning diffusion generative-models human-motion-generation human-motion-prediction iccv2023 latent-diffusion ldm motion-forecasting motion-prediction pose-forecasting pose-prediction pytorch

Last synced: 10 Nov 2024

https://github.com/dansuh17/segan-pytorch

SEGAN pytorch implementation https://arxiv.org/abs/1703.09452

audio data-preprocessing mir pytorch segan segan-pytorch source-separation speech-enhancement

Last synced: 01 Nov 2024

https://github.com/gordicaleksa/pytorch-neural-style-transfer-johnson

Reconstruction of the fast neural style transfer (Johnson et al.). Some portions of the paper have been improved by the follow-up work like the instance normalization, etc. Checkout transformer_net.py's header for details.

deep-learning deep-learning-tutorial machine-learning neural-style-transfer neural-style-transfer-tutorial non-photorealistic-rendering python pytorch style-transfer

Last synced: 11 Nov 2024

https://github.com/dongminlee94/samsung-drl-code

A repository for implementation of deep reinforcement learning lectured at Samsung

deep-reinforcement-learning model-free-rl pytorch pytorch-rl reinforcement-learning

Last synced: 12 Nov 2024

https://github.com/V2AI/EFG

An Efficient, Flexible, and General deep learning framework that retains minimal.

3d-object-detection 3d-scene-understanding detr nuscenes object-detection pytorch waymo-open-dataset

Last synced: 28 Oct 2024

https://github.com/aredden/flux-fp8-api

Flux diffusion model implementation using quantized fp8 matmul & remaining layers use faster half precision accumulate, which is ~2x faster on consumer devices.

diffusion fast-inference flux fp8 pytorch quantization

Last synced: 20 Sep 2024

https://github.com/xuarehere/yolo_series_deepsort_pytorch

Deepsort with yolo series. This project support the existing yolo detection model algorithm (YOLOV8, YOLOV7, YOLOV6, YOLOV5, YOLOV4Scaled, YOLOV4, YOLOv3', PPYOLOE, YOLOR, YOLOX ).

deep-sort deep-sort-tracking deepsort mot mot-tracking object-tracking pytorch sort yolo yolov3 yolov4 yolov4-deepsort yolov5 yolov5-deepsort yolov5-deepsort-pytorch yolov6 yolov6-deepsort yolov7 yolov7-deepsort yolov8-deepsort

Last synced: 09 Nov 2024

https://github.com/okrasolar/pytorch-timeseries

PyTorch implementations of neural networks for timeseries classification

classification deep-learning machine-learning pytorch time-series

Last synced: 11 Nov 2024

https://github.com/OxWearables/ssl-wearables

Self-supervised learning for wearables using the UK-Biobank (>700,000 person-days)

accelerometer deep-learning foundation-models human-activity-recognition pytorch self-supervised-learning wearable

Last synced: 06 Nov 2024

https://github.com/wassname/viz_torch_optim

Videos of deep learning optimizers moving on 3D problem-landscapes

3d-graphics deep-learning learning-rate machine-learning optimization-algorithms optimizer pytorch vizualisation

Last synced: 15 Nov 2024

https://github.com/newlei/LR-GCCF

Revisiting Graph based Collaborative Filtering: A Linear Residual Graph Convolutional Network Approach, AAAI2020

aaai2020 collaborative-filtering graph-based-recommendation graph-neural-networks pytorch recommendation

Last synced: 08 Aug 2024

https://github.com/kdexd/lang-emerge-parlai

Implementation of EMNLP 2017 Paper "Natural Language Does Not Emerge 'Naturally' in Multi-Agent Dialog" using PyTorch and ParlAI

emnlp language-grounding parlai pytorch reinforcement-learning

Last synced: 09 Nov 2024

https://github.com/Little-Podi/AiATrack

[ECCV'22] The official PyTorch implementation of our ECCV 2022 paper: "AiATrack: Attention in Attention for Transformer Visual Tracking".

attention-mechanism eccv2022 object-tracking pytorch single-object-tracking tracking vision-transformer visual-tracking

Last synced: 02 Nov 2024

https://github.com/knodle/knodle

A PyTorch-based open-source framework that provides methods for improving the weakly annotated data and allows researchers to efficiently develop and compare their own methods.

ai classification denoising-methods knodle machine-learning natural-language-procressing python pytorch relation-extraction snorkel weak-supervision weakly-supervised-learning

Last synced: 26 Oct 2024

https://github.com/allentdan/libtorchdetection

C++ trainable detection library based on libtorch (or pytorch c++). Yolov4 tiny provided now.

cpp detection detection-model libtorch pretrained pretrained-weights pytorch pytorch-cpp yolo yolov4 yolov4-tiny

Last synced: 27 Oct 2024

https://github.com/jmisilo/clip-gpt-captioning

CLIPxGPT Captioner is Image Captioning Model based on OpenAI's CLIP and GPT-2.

computer-vision cv deep-learning image-caption image-caption-generator image-captioning machine-learning nlp python pytorch

Last synced: 04 Nov 2024

https://github.com/ziwei-zh/CorrPM

Pytorch implementation of CVPR2020 paper "Correlating Edge, Pose with Parsing"

cvpr cvpr2020 human-parsing pytorch semantic-segmantation

Last synced: 08 Aug 2024

https://github.com/moskomule/homura

homura is a library for fast prototyping DL research

pytorch utility

Last synced: 13 Oct 2024

https://github.com/Jingliang-Duan/DSAC-v1

DSAC; Distributional Soft Actor-Critic

pytorch reinforcement-learning

Last synced: 08 Aug 2024

https://github.com/TristanBilot/mlx-benchmark

Benchmark of Apple MLX operations on all Apple Silicon chips (GPU, CPU) + MPS and CUDA.

apple-silicon benchmark deep-learning machine-learning mlx pytorch

Last synced: 06 Aug 2024

https://github.com/d-li14/hbonet

[ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions, surpassing MobileNetV2

efficient-model iccv2019 imagenet mobilenetv2 pretrained-models pytorch

Last synced: 07 Nov 2024

https://github.com/cscfi/machine-learning-scripts

Collection of scripts and tools related to machine learning

jupyter-notebooks keras machine-learning notebooks pytorch scikit-learn tensorflow

Last synced: 10 Nov 2024

https://github.com/xpixelgroup/rethinkvsralignment

(NIPS 2022) Rethinking Alignment in Video Super-Resolution Transformers

nips-2022 pytorch super-resolution vision-transformer

Last synced: 11 Nov 2024

https://github.com/AiAiHealthcare/ProjectAiAi

AiAi.care project is teaching computers to "see" chest X-rays and interpret them how a human Radiologist would. We are using 700,000 Chest X-Rays + Deep Learning to build an FDA 💊 approved, open-source screening tool for Tuberculosis and Lung Cancer. After an MRMC clinical trial, AiAi CAD will be distributed for free to emerging nations, charitable hospitals, and organizations like WHO 🌏 We will also release our pretrained models and weights as Medical Imagenet.

charity convolutional-neural-networks ct-scans cxr-lungs deep-learning dicom fda keras lung-cancer lung-cancer-detection postero-anterior pytorch radiologist radiology scikit-image teleradiology tensorflow tuberculosis x-ray

Last synced: 04 Aug 2024

https://github.com/EIDOSLAB/torchstain

Stain normalization tools for histological analysis and computational pathology

computational-pathology digital-pathology histopathology medical-imaging numpy python pytorch stain-normalization tensorflow

Last synced: 13 Nov 2024

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

[ICML2020] "AutoGAN-Distiller: Searching to Compress Generative Adversarial Networks" by Yonggan Fu, Wuyang Chen, Haotao Wang, Haoran Li, Yingyan Lin, Zhangyang Wang

compression generative-adversarial-network neural-architecture-search pytorch

Last synced: 16 Nov 2024

https://github.com/xudejing/video-clip-order-prediction

Self-supervised Spatiotemporal Learning via Video Clip Order Prediction

cvpr2019 pytorch self-supervised-learning spatiotemporal video

Last synced: 11 Nov 2024

https://github.com/funcwj/uPIT-for-speech-separation

Speech separation with utterance-level PIT experiments

pit pytorch speech-separation

Last synced: 02 Nov 2024

https://github.com/svip-lab/GazeFollowing

Code for ACCV2018 paper 'Believe It or Not, We Know What You Are Looking at!'

accv2018 gaze-follow pytorch

Last synced: 31 Oct 2024

https://github.com/fkodom/yet-another-retnet

A simple but robust PyTorch implementation of RetNet from "Retentive Network: A Successor to Transformer for Large Language Models" (https://arxiv.org/pdf/2307.08621.pdf)

deep-learning llms machine-learning natural-language-processing neural-networks python pytorch

Last synced: 28 Oct 2024

https://github.com/XuyangBai/PPF-FoldNet

PyTorch reimplementation for "PPF-FoldNet: Unsupervised Learning of Rotation Invariant 3D Local Descriptors" https://arxiv.org/abs/1808.10322

point-cloud ppf-foldnet ppfnet pytorch registration

Last synced: 28 Oct 2024

https://github.com/paganpasta/eqxvision

A Python package of computer vision models for the Equinox ecosystem.

equinox python pytorch vision

Last synced: 02 Nov 2024

https://github.com/pytorch-labs/torchfix

TorchFix - a linter for PyTorch-using code with autofix support

flake8 flake8-plugin hacktoberfest linter python pytorch static-analysis static-code-analysis

Last synced: 15 Nov 2024

https://github.com/lonePatient/BERT-chinese-text-classification-pytorch

This repo contains a PyTorch implementation of a pretrained BERT model for text classification.

bert chinese chinese-text-classification nlp pytorch text-classification

Last synced: 02 Nov 2024