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/kefirski/pytorch_NEG_loss

NEG loss implemented in pytorch

python pytorch word2vec

Last synced: 03 Aug 2024

https://github.com/miccunifi/SEARLE

[ICCV 2023] - Zero-shot Composed Image Retrieval with Textual Inversion

circo cirr clip composed-image-retrieval fashion-iq knowledge-distillation multimodal-learning pytorch textual-inversion

Last synced: 01 Aug 2024

https://github.com/ami-iit/adam

adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.

adam-robotics automatic-differentiation casadi jax motion-planning numpy optimization python pytorch reinforcement-learning rigid-body-dynamics robotics urdf

Last synced: 25 Sep 2024

https://github.com/emilemathieu/pvae

code for "Continuous Hierarchical Representations with Poincaré Variational Auto-Encoders".

hierarchical-data hyperbolic-geometry poincare-embeddings pytorch vae

Last synced: 25 Sep 2024

https://github.com/jingtaozhan/DRhard

SIGIR'21: Optimizing DR with hard negatives and achieving SOTA first-stage retrieval performance on TREC DL Track.

information-retrieval pytorch web-search

Last synced: 03 Aug 2024

https://github.com/zhiqwang/sightseq

Computer vision tools for fairseq, containing PyTorch implementation of text recognition and object detection

attention crnn ctc densenet faster-rcnn image-captioning mobilenet object-detection ocr pytorch scene-texts text-recognition transformer

Last synced: 07 Aug 2024

https://github.com/Jasonkks/PTTR

Pytorch Implementation of PTTR: Relational 3D Point Cloud Object Tracking with Transformer

autonomous-driving point-cloud pytorch tracking

Last synced: 31 Jul 2024

https://github.com/borchero/pycave

Traditional Machine Learning Models for Large-Scale Datasets in PyTorch.

gaussian-mixture-models kmeans machine-learning markov-model pytorch pytorch-lightning

Last synced: 02 Aug 2024

https://github.com/leverxgroup/esrgan

Enhanced SRGAN. Champion PIRM Challenge on Perceptual Super-Resolution

computer-vision deep-neural-networks generative-adversarial-network image-processing pytorch super-resolution

Last synced: 30 Jul 2024

https://github.com/rsnk96/Ubuntu-Setup-Scripts

Scripts to help you set up your Ubuntu quickly, especially if you're in any subfield of Data Science or AI!

anaconda cuda deep-learning deeplearning dl ffmpeg installers ml opencv python pytorch tensorflow tensorflow-setup ubuntu zsh

Last synced: 01 Aug 2024

https://github.com/developmentseed/fastai-serving

A Docker image for serving fast.ai models, mimicking the API of Tensorflow Serving

deep-learning docker machine-learning python pytorch

Last synced: 05 Aug 2024

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

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

chinese-ner ner pytorch tplinker tplinker-ner

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

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

[CVPR'22] Official PyTorch implementation of Integrative Few-Shot Learning for Classification and Segmentation

computer-vision cvpr2022 deep-learning few-shot-classification few-shot-learning few-shot-segmentation pytorch pytorch-lightning

Last synced: 03 Aug 2024

https://github.com/astorfi/sequence-to-sequence-from-scratch

:speech_balloon: Sequence to Sequence from Scratch Using Pytorch

deep-learning machine-learning python pytorch

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

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

https://github.com/nmichlo/disent

🧶 Modular VAE disentanglement framework for python built with PyTorch Lightning ▸ Including metrics and datasets ▸ With strongly supervised, weakly supervised and unsupervised methods ▸ Easily configured and run with Hydra config ▸ Inspired by disentanglement_lib

autoencoders configurable datasets disentangled-representations disentanglement metric-learning metrics python python3 pytorch pytorch-lightning representation-learning vae

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

https://github.com/SherylHYX/pytorch_geometric_signed_directed

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for PyTorch Geometric. The paper is accepted by LoG 2023.

deep-learning directed-networks gnn graph-neural-netowrks machine-learning networks python pytorch pytorch-geometric signed-networks

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

https://github.com/yurizzzzz/Helmet-Detection-YoloV5

Safety helmet wearing detection on construction site based on YoloV5s-V5.0 including helmet dataset(基于YoloV5-V5.0的工地安全帽检测并且包含开源的安全帽数据集)

python pytorch yolov5s

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

https://github.com/stepelu/ptstat

Probabilistic Programming and Statistical Inference in PyTorch

probability pytorch statistics

Last synced: 03 Aug 2024

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

PyTorch implementation of Res2Net

deep-learning pytorch res2net

Last synced: 31 Jul 2024

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

deep clustering method for single-channel speech separation

pytorch speech-separation

Last synced: 01 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: 03 Aug 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: 03 Aug 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 Jul 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: 02 Aug 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: 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: 02 Aug 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/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/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/okrasolar/pytorch-timeseries

PyTorch implementations of neural networks for timeseries classification

classification deep-learning machine-learning pytorch time-series

Last synced: 02 Aug 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: 03 Aug 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: 31 Jul 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: 30 Jul 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: 25 Sep 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/Jingliang-Duan/DSAC-v1

DSAC; Distributional Soft Actor-Critic

pytorch reinforcement-learning

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

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

Speech separation with utterance-level PIT experiments

pit pytorch speech-separation

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

https://github.com/antoinebrl/torchextractor

Feature extraction made simple with torchextractor

feature-extraction machine-learning python3 pytorch

Last synced: 24 Sep 2024