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/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/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/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/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/stepelu/ptstat

Probabilistic Programming and Statistical Inference in PyTorch

probability pytorch statistics

Last synced: 03 Aug 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/4uiiurz1/pytorch-res2net

PyTorch implementation of Res2Net

deep-learning pytorch res2net

Last synced: 31 Jul 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/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/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/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/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/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/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/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/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/moskomule/homura

homura is a library for fast prototyping DL research

pytorch utility

Last synced: 03 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/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/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/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/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/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/Jingliang-Duan/DSAC-v1

DSAC; Distributional Soft Actor-Critic

pytorch reinforcement-learning

Last synced: 08 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/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/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/xudejing/video-clip-order-prediction

Self-supervised Spatiotemporal Learning via Video Clip Order Prediction

cvpr2019 pytorch self-supervised-learning spatiotemporal video

Last synced: 02 Aug 2024

https://github.com/antoinebrl/torchextractor

Feature extraction made simple with torchextractor

feature-extraction machine-learning python3 pytorch

Last synced: 01 Aug 2024

https://github.com/osai-ai/dokai

Collection of Docker images for ML/DL and video processing projects

cuda deep-learning docker docker-image ffmpeg opencv python pytorch tensorrt video-processing

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

https://github.com/ritchieng/eigenvectors-from-eigenvalues

PyTorch implementation comparison of old and new method of determining eigenvectors from eigenvalues.

colab determining-eigenvectors eigenvalues eigenvectors pytorch

Last synced: 30 Jul 2024

https://github.com/mzaradzki/factorization-machine-for-prediction

Factorization Machine for regression and classification

keras machine-learning python pytorch theano

Last synced: 10 Aug 2024

https://github.com/anibali/margipose

3D monocular human pose estimation

deep-learning pose-estimation pytorch

Last synced: 08 Aug 2024

https://github.com/paganpasta/eqxvision

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

equinox python pytorch vision

Last synced: 01 Aug 2024

https://github.com/openvinotoolkit/openvino_contrib

Repository for OpenVINO's extra modules

arm inference-engine java nvidia-gpu openvino pytorch

Last synced: 31 Jul 2024

https://github.com/chail/gan-ensembling

Invert and perturb GAN images for test-time ensembling

generative-adversarial-network image-classfication image-synthesis pytorch

Last synced: 03 Aug 2024

https://github.com/JinheonBaek/GMT

Official Code Repository for the paper "Accurate Learning of Graph Representations with Graph Multiset Pooling" (ICLR 2021)

graph graph-classification graph-multiset-pooling graph-multiset-transformer graph-neural-networks graph-pooling graph-reconstruction graph-representation graph-representation-learning graph-transformer pytorch

Last synced: 08 Aug 2024

https://github.com/kunhe/FastAP-metric-learning

Code for CVPR 2019 paper "Deep Metric Learning to Rank"

cvpr deep-metric-learning learning-to-rank matconvnet metric-learning pytorch

Last synced: 03 Aug 2024

https://github.com/chenyuntc/scene-baseline

PyTorch baseline for AI challenger Scene classification

image-classification pytorch

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

https://github.com/ajhamdi/MVTN

pytorch implementation of the ICCV'21 paper "MVTN: Multi-View Transformation Network for 3D Shape Recognition"

3d 3d-models classification deep-learning iccv2021 point-cloud pytorch

Last synced: 31 Jul 2024

https://github.com/eric-ai-lab/PEViT

Official implementation of AAAI 2023 paper "Parameter-efficient Model Adaptation for Vision Transformers"

fine-tuning image-classification parameter-efficient-tuning pytorch vision-transformer

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

https://github.com/doocs/deep-learning

🙃 深度学习实践与知识总结

ai deep-learning deep-neural-networks doocs pytorch tensorflow

Last synced: 31 Jul 2024

https://github.com/eveningglow/multitask-CycleGAN

Pytorch implementation of multitask CycleGAN with auxiliary classification loss

computer-vision cyclegan deep-learning gan generative-adversarial-network image-translation pytorch

Last synced: 03 Aug 2024

https://github.com/jkoutsikakis/pytorch-wrapper

Provides a systematic and extensible way to build, train, evaluate, and tune deep learning models using PyTorch.

data-science deep-learning machine-learning neural-network python pytorch pytorch-wrapper tensor

Last synced: 07 Aug 2024

https://github.com/nitin-rathi/hybrid-snn-conversion

Training spiking networks with hybrid ann-snn conversion and spike-based backpropagation

ann-snn-conversion backpropagation-algorithm deep-learning pytorch snn spiking-neural-networks

Last synced: 02 Aug 2024

https://github.com/wchliao/multi-task-image-classification

Multi-task learning for image classification implemented in PyTorch.

cifar-100 cifar10 deep-learning machine-learning multi-task-learning omniglot pytorch

Last synced: 03 Aug 2024

https://github.com/xunge/pytorch_lmdb_imagenet

Use lmdb to speed up imagenet dataset

imagenet lmdb pytorch

Last synced: 03 Aug 2024

https://github.com/takuseno/minerva

An out-of-the-box GUI tool for offline deep reinforcement learning

deep-learning deep-reinforcement-learning offline-rl pytorch

Last synced: 02 Aug 2024

https://github.com/zhiyiYo/Alpha-Gobang-Zero

A gobang robot based on reinforcement learning.

alphazero gobang pyqt5 pytorch reinforcement-learning

Last synced: 02 Aug 2024

https://github.com/hpcaitech/CachedEmbedding

A memory efficient DLRM training solution using ColossalAI

colossal-ai deep-learning dlrm embeddings nlp pytorch recommandation-system

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

https://github.com/sophiefy/StellaVoiceChanger

Deep-learning-based voice changer, supporting local inference.

deep-learning python pytorch voice-conversion

Last synced: 03 Aug 2024

https://github.com/Torsion-Audio/nn-inference-template

Neural network inference template for real-time cricital audio environments - presented at ADC23

audio audio-plugin deep-learning inference inference-engine juce libtorch machine-learning neural-network onnx onnxruntime pytorch tensorflow tensorflow-lite

Last synced: 03 Aug 2024

https://github.com/lxc86739795/human_vehicle_parsing_platform

A pytorch codebase for human parsing and vehicle parsing

codebase human parsing person pytorch segmentation vehicle vehicleid veri-wild veri776

Last synced: 01 Aug 2024

https://github.com/huangtinglin/MixGCF

MixGCF: An Improved Training Method for Graph Neural Network-based Recommender Systems, KDD2021

graph-neural-network information-retrieval negative-sampling network-embedding pytorch recommender-system

Last synced: 08 Aug 2024

https://github.com/LBM-EPFL/PeSTo

Geometric deep learning method to predict protein binding interfaces from a protein structure.

deep-learning geometric-deep-learning protein-binding-site protein-protein-interface protein-structure pytorch

Last synced: 02 Aug 2024

https://github.com/sberbank-ai-lab/pytorch-lifestream

A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision

machine-learning pytorch self-supervised-learning

Last synced: 07 Aug 2024

https://github.com/roman-vygon/triplet_loss_kws

Learning Efficient Representations for Keyword Spotting with Triplet Loss

deep-learning keyword-spotting pytorch speech-recognition

Last synced: 01 Aug 2024

https://github.com/alldbi/SuperMix

Pytorch implementation of CVPR2021 paper: SuperMix: Supervising the Mixing Data Augmentation

augmentation cvpr2021 deep-learning distillation pytorch saliency-detection supervised

Last synced: 02 Aug 2024

https://github.com/drivendataorg/zamba

A Python package for identifying 42 kinds of animals, training custom models, and estimating distance from camera trap videos

animals camera-traps chimps cli conservation deep-learning ecology gpu jungle machine-learning neural-network python pytorch pytorch-lightning video-processing videos

Last synced: 01 Aug 2024

https://github.com/Coderx7/SimpleNet

This repository contains the architectures, Models, logs, etc pertaining to the SimpleNet Paper (Lets keep it simple: Using simple architectures to outperform deeper architectures )

cnn-model cnn-pytorch convolutional-neural-networks deep-learning fast-net image-classification imagenet pytorch simplenet

Last synced: 01 Aug 2024

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

[NeurIPS2022] This is the official code of "CAGroup3D: Class-Aware Grouping for 3D Object Detection on Point Clouds".

3d-object-detection pytorch scannet-dataset sunrgbd

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

PyTorch Awesome Lists
Awesome-pytorch-list 704 awesome-semantic-segmentation-pytorch 35 Awesome-pytorch-list-CNVersion 692 CV-pretrained-model 103 awesome-colab-notebooks 502 awesome-AutoML-and-Lightweight-Models 150 Awesome-Denoise 164 ai4artists 286 arctic 34 awesome-yolov8-models 43 awesome-cuda-and-hpc 424 awesome-python-machine-learning-resources 1,884 awesome-adaptive-computation 171 awesome-NILM-with-code 47 awesome-dotnet-machine-learning 134 awesome-Multi-Document-Summarization 56 PyTorch-Guide 616 awesome-fastai 42 awesome-ai-tools 499
PyTorch Categories
Pytorch & related libraries 228 Pytorch & related libraries｜Pytorch & 相关库 218 Research 197 Tutorials 192 文本数据和NLP 182 Blogs 164 Uncategorized 161 Frameworks 142 数据可视化 130 机器学习框架 123 1.) Neural Architecture Search 113 图像数据与CV 105 Learning 102 Blogposts :sparkle: 101 模型的可解释性 100 Interactive Learning & Tools 96 超参数优化和AutoML 92 Tutorials, books, & examples 82 Tutorials & books & examples｜教程 & 书籍 & 示例 80 工作流程和实验跟踪 80