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/lartpang/ZoomNet

Zoom In and Out: A Mixed-scale Triplet Network for Camouflaged Object Detection, CVPR 2022

camouflaged-object-detection cod codeforpaper cvpr cvpr2022 paper papercode python pytorch rgbcod rgbsod saliency-detection salient-object-detection sod

Last synced: 15 Nov 2024

https://github.com/yeyupiaoling/pytorch-mobilefacenet

Pytorch实现的人脸识别明细MobileFaceNet模型,在预测使用MTCNN检测人脸,然后使用MobileFaceNet模型识别。

face-recognition mobilefacenet mtcnn pytorch

Last synced: 01 Nov 2024

https://github.com/gaohongkui/TPLinker-NER

TPlinker for NER 中文/英文命名实体识别

chinese-ner ner pytorch tplinker tplinker-ner

Last synced: 06 Nov 2024

https://github.com/yuweihao/kern

Code for Knowledge-Embedded Routing Network for Scene Graph Generation (CVPR 2019)

computer-vision graph-neural-network pytorch scene-graph scene-graph-generation

Last synced: 06 Nov 2024

https://github.com/xuyxu/deep-clustering-network

PyTorch Implementation of "Towards K-Means-Friendly Spaces: Simultaneous Deep Learning and Clustering," Bo Yang et al., ICML'2017.

clustering deep-learning pytorch

Last synced: 08 Nov 2024

https://github.com/yuweihao/KERN

Code for Knowledge-Embedded Routing Network for Scene Graph Generation (CVPR 2019)

computer-vision graph-neural-network pytorch scene-graph scene-graph-generation

Last synced: 02 Nov 2024

https://github.com/cceyda/torchserve-dashboard

Management Dashboard for Torchserve

ml-deployment pytorch streamlit torchserve

Last synced: 31 Oct 2024

https://github.com/marcoslucianops/deepstream-yolo-pose

NVIDIA DeepStream SDK 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 application for YOLO-Pose models

deepstream nvidia nvidia-deepstream-sdk object-detection pose-estimation pytorch tensorrt ultralytics yolo yolov8

Last synced: 16 Nov 2024

https://github.com/JusperLee/Deep-Clustering-for-Speech-Separation

Pytorch implements Deep Clustering: Discriminative Embeddings For Segmentation And Separation

deep-clustering pytorch segmentation speech-separation

Last synced: 02 Nov 2024

https://github.com/amazon-science/tanl

Structured Prediction as Translation between Augmented Natural Languages

deep-learning natural-language-processing pytorch

Last synced: 12 Nov 2024

https://github.com/lonepatient/bilstm-crf-ner-pytorch

This repo contains a PyTorch implementation of a BiLSTM-CRF model for named entity recognition task.

bilstm-crf crf lstm ner nlp pytorch

Last synced: 06 Nov 2024

https://github.com/graykode/toeicbert

TOEIC(Test of English for International Communication) solving using pytorch-pretrained-BERT model.

ai bert deep-learning lm mask nlp pytorch pytorch-pretrained toeic

Last synced: 01 Nov 2024

https://github.com/kedreamix/pytorch-image-classification

用于pytorch的图像分类,包含多种模型方法,比如AlexNet,VGG,GoogleNet,ResNet,DenseNet等等,包含可完整运行的代码。除此之外,也有colab的在线运行代码,可以直接在colab在线运行查看结果。也可以迁移到自己的数据集进行迁移学习。

alexnet cifar10 densenet goolenet inception mnasnet mobilenet pytorch resnet shufflenet vgg

Last synced: 17 Nov 2024

https://github.com/qq44642754a/Yolov5_ros

Real-time object detection with ROS, based on YOLOv5 and PyTorch (基于 YOLOv5的ROS实时对象检测)

pytorch ros yolov5

Last synced: 09 Nov 2024

https://github.com/vra/flopth

A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.

cnn deep-learning deeplearning flop flops pip pypi python python3 pytorch visualization

Last synced: 14 Nov 2024

https://github.com/nvidia/clara-train-examples

Example notebooks demonstrating how to use Clara Train to build Medical Imaging Deep Learning models

automl clara-train deep-learning healthcare-imaging medical-imaging-computing medical-imaging-processing python3 pytorch tcia-dac

Last synced: 29 Oct 2024

https://github.com/lupantech/intergps

Data and Code for ACL 2021 Paper "Inter-GPS: Interpretable Geometry Problem Solving with Formal Language and Symbolic Reasoning"

mathqa pytorch

Last synced: 05 Nov 2024

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

an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors

cuda cupy deep-learning python pytorch

Last synced: 14 Nov 2024

https://github.com/chrhenning/hypnettorch

Package for working with hypernetworks in PyTorch.

deep-learning hypernet hypernets hypernetwork hypernetworks neural-network python pytorch torch

Last synced: 11 Nov 2024

https://github.com/ise-uiuc/nnsmith

Automatic DNN generation for fuzzing and more

compiler deep-learning fuzzing machine-learning pytorch tensorflow

Last synced: 11 Oct 2024

https://github.com/um-arm-lab/pytorch_volumetric

Volumetric structures such as voxels and SDFs implemented in pytorch

chamfer-distance kinematics pytorch robot sdf signed-distance-field signed-distance-functions voxel voxel-grid voxels

Last synced: 16 Nov 2024

https://github.com/AlbertSuarez/object-cut

✂️ Cut the main object of an image automagically

basnet connexion crop-image image-processing openapi pytorch

Last synced: 26 Oct 2024

https://github.com/lucidrains/pytorch-custom-utils

Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new AI research

accelerate pytorch

Last synced: 12 Nov 2024

https://github.com/bradyfu/dvg

[NeurIPS 2019] Dual Variational Generation for Low Shot Heterogeneous Face Recognition

face-recognition heterogeneous-face-recognition pytorch

Last synced: 16 Nov 2024

https://github.com/alibaba/graphlearn-for-pytorch

A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.

deep-learning distributed gpu graph-neural-networks pytorch

Last synced: 12 Nov 2024

https://github.com/bharathgs/nalu

Basic pytorch implementation of NAC/NALU from Neural Arithmetic Logic Units paper by trask et.al

accumulator alu arithmetic deep-learning deepmind logic machine-learning nalu neural-network pytorch

Last synced: 15 Nov 2024

https://github.com/ajayarunachalam/msda

Library for multi-dimensional, multi-sensor, uni/multivariate time series data analysis, unsupervised feature selection, unsupervised deep anomaly detection, and prototype of explainable AI for anomaly detector

anamoly-detection-using-graphs anomaly-detection correlation data-analysis deep-learning deep-neural-networks explainable-artificial-intelligence feature-engineering feature-selection multidimensional-data multisensor python pytorch sensor sensor-data signal-processing tabular-data time-series variation visualization

Last synced: 06 Nov 2024

https://github.com/bharathgs/NALU

Basic pytorch implementation of NAC/NALU from Neural Arithmetic Logic Units paper by trask et.al

accumulator alu arithmetic deep-learning deepmind logic machine-learning nalu neural-network pytorch

Last synced: 14 Nov 2024

https://github.com/snakers4/russian_stt_text_normalization

Russian text normalization pipeline for speech-to-text and other applications based on tagging s2s networks

python3 pytorch russian-language speech speech-to-text text-normalization torchscript

Last synced: 07 Aug 2024

https://github.com/elliottzheng/face-detection

Fast and reliable face detection with RetinaFace.PyTorch

face-detection python pytorch retinaface

Last synced: 27 Oct 2024

https://github.com/yanndubs/lossyless

Generic image compressor for machine learning. Pytorch code for our paper "Lossy compression for lossless prediction".

compression deep-learning machine-learning python pytorch self-supervised-learning

Last synced: 27 Oct 2024

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/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/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/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/pytorch/extension-script

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

cpp cuda pytorch

Last synced: 26 Sep 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/senya-ashukha/simple-equivariant-gnn

A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks

deep-learning deep-neural-networks egnn equivariance gnn graph-neural-networks homo molecules neural-networks pytorch qm9

Last synced: 19 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/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/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/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/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/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/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: 18 Nov 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/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/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/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/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/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/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/kuangliu/pytorch-ssd

Single Shot MultiBox Detector in PyTorch [deprecated]

pytorch ssd

Last synced: 10 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/funcwj/deep-clustering

deep clustering method for single-channel speech separation

pytorch speech-separation

Last synced: 02 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/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/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/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/stepelu/ptstat

Probabilistic Programming and Statistical Inference in PyTorch

probability pytorch statistics

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