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/shibing624/pytextclassifier

pytextclassifier is a toolkit for text classification. 文本分类,LR,Xgboost,TextCNN,FastText,TextRNN,BERT等分类模型实现,开箱即用。

bert classification focalloss-pytorch hierarchical machine-learning nlp pytextclassifier python pytorch softmax text-classification text-classifier

Last synced: 11 Oct 2024

https://github.com/adambielski/CapsNet-pytorch

PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules

capsnet capsules deep-learning dynamic-routing-between-capsules machine-learning mnist pytorch

Last synced: 07 Aug 2024

https://github.com/jaketae/storyteller

Multimodal AI Story Teller, built with Stable Diffusion, GPT, and neural text-to-speech

ddpm diffusion-models gpt image-generation natural-language-generation pytorch stable-diffusion text-to-image text-to-speech text-to-video video-generation

Last synced: 04 Aug 2024

https://github.com/wvangansbeke/sparse-depth-completion

Predict dense depth maps from sparse and noisy LiDAR frames guided by RGB images. (Ranked 1st place on KITTI) [MVA 2019]

computer-vision deep-learning depth-completion depth-prediction kitti lidar noisy-data pytorch sensor-fusion

Last synced: 14 Nov 2024

https://github.com/crowsonkb/style-transfer-pytorch

Neural style transfer in PyTorch.

neural-art pytorch style-transfer

Last synced: 13 Nov 2024

https://github.com/nv-tlabs/steal

STEAL - Learning Semantic Boundaries from Noisy Annotations (CVPR 2019)

annotation cvpr2019 deep-learning devil-is-in-the-edges nv-tlabs pytorch semantic-boundaries semantic-segmentation steal

Last synced: 15 Nov 2024

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

[ICLR24] Official implementation of the paper “MagicDrive: Street View Generation with Diverse 3D Geometry Control”

autonomous-vehicles deep-learning diffusion-models image-generation pytorch video-generation

Last synced: 28 Oct 2024

https://github.com/ma-xu/pointMLP-pytorch

[ICLR 2022 poster] Official PyTorch implementation of "Rethinking Network Design and Local Geometry in Point Cloud: A Simple Residual MLP Framework"

modelnet40 pointcloud pytorch scanobjectnn

Last synced: 28 Oct 2024

https://github.com/fkodom/fft-conv-pytorch

Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.

convolution image-processing neural-networks python3 pytorch

Last synced: 28 Oct 2024

https://github.com/wildltr/ptranking

Learning to Rank in PyTorch

learning-to-rank pytorch ranking

Last synced: 11 Nov 2024

https://github.com/archinetai/surgeon-pytorch

A library to inspect and extract intermediate layers of PyTorch models.

artificial-intelligence deep-learning pytorch

Last synced: 05 Nov 2024

https://github.com/Neutone/neutone_sdk

Join the community on Discord for more discussions around Neutone! https://discord.gg/VHSMzb8Wqp

ai audio audio-plugin deep-learning machine-learning music python pytorch realtime-audio sdk vst

Last synced: 05 Nov 2024

https://github.com/kevinzakka/recurrent-visual-attention

A PyTorch Implementation of "Recurrent Models of Visual Attention"

attention pytorch ram recurrent-attention-model recurrent-models

Last synced: 30 Oct 2024

https://github.com/ZwwWayne/K-Net

[NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation

image-segmentation mmdetection mmsegmentation panoptic-segmentation pytorch semantic-segmentation

Last synced: 15 Nov 2024

https://github.com/tri-ml/dd3d

Official PyTorch implementation of DD3D: Is Pseudo-Lidar needed for Monocular 3D Object detection? (ICCV 2021), Dennis Park*, Rares Ambrus*, Vitor Guizilini, Jie Li, and Adrien Gaidon.

computer-vision deep-learning pytorch

Last synced: 13 Nov 2024

https://github.com/zwwwayne/k-net

[NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation

image-segmentation mmdetection mmsegmentation panoptic-segmentation pytorch semantic-segmentation

Last synced: 09 Nov 2024

https://github.com/JingyibySUTsoftware/Yolov5-deepsort-driverDistracted-driving-behavior-detection

基于深度学习的驾驶员分心驾驶行为(疲劳+危险行为)预警系统使用YOLOv5+Deepsort实现驾驶员的危险驾驶行为的预警监测

deeplearning deepsort detection dlib dlib-face-detection dlib-tracker driver-behavior object-detection opencv playsound pyqt5 python pytorch sort tired tracker yolov5 yolov5-deepsort

Last synced: 09 Nov 2024

https://github.com/backtime92/CRAFT-Reimplementation

CRAFT-Pyotorch:Character Region Awareness for Text Detection Reimplementation for Pytorch

craft pytorch text-detection

Last synced: 03 Nov 2024

https://github.com/microsoft/archai

Accelerate your Neural Architecture Search (NAS) through fast, reproducible and modular research.

automated-machine-learning automl darts deep-learning hyperparameter-optimization machine-learning model-compression nas neural-architecture-search petridish python pytorch

Last synced: 07 Oct 2024

https://github.com/chingyaoc/ggnn.pytorch

A PyTorch Implementation of Gated Graph Sequence Neural Networks (GGNN)

ggnn gnns graph pytorch

Last synced: 30 Oct 2024

https://github.com/ahangchen/torch_base

Quickly bring up your PyTorch project(a skeleton)

computer-vision deep-learning machine-learning pytorch

Last synced: 11 Nov 2024

https://github.com/criteo-research/reco-gym

Code for reco-gym: A Reinforcement Learning Environment for the problem of Product Recommendation in Online Advertising

criteo openai-gym pytorch recommendation-system

Last synced: 03 Aug 2024

https://github.com/davidefiocco/streamlit-fastapi-model-serving

Simple web app example serving a PyTorch model using streamlit and FastAPI

docker-compose fastapi pycones pytorch streamlit

Last synced: 13 Oct 2024

https://github.com/wenmuzhou/psenet.pytorch

A pytorch re-implementation of PSENet: Shape Robust Text Detection with Progressive Scale Expansion Network

ocr python3 pytorch text-detection

Last synced: 14 Nov 2024

https://github.com/mit-han-lab/mcunet

[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning

deep-learning microncontroller neural-architecture-search pytorch tinyml

Last synced: 28 Oct 2024

https://github.com/salesforce/warp-drive

Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)

cuda deep-learning gpu high-throughput multiagent-reinforcement-learning numba pytorch reinforcement-learning

Last synced: 14 Nov 2024

https://github.com/TRI-ML/dd3d

Official PyTorch implementation of DD3D: Is Pseudo-Lidar needed for Monocular 3D Object detection? (ICCV 2021), Dennis Park*, Rares Ambrus*, Vitor Guizilini, Jie Li, and Adrien Gaidon.

computer-vision deep-learning pytorch

Last synced: 28 Oct 2024

https://github.com/edisonleeeee/graphgallery

GraphGallery is a gallery for benchmarking Graph Neural Networks, From InplusLab.

adversarial-machine-learning dgl geometric-deep-learning graph graph-neural-networks graphgallery models pyg pytorch

Last synced: 13 Nov 2024

https://github.com/EdisonLeeeee/GraphGallery

GraphGallery is a gallery for benchmarking Graph Neural Networks, From InplusLab.

adversarial-machine-learning dgl geometric-deep-learning graph graph-neural-networks graphgallery models pyg pytorch

Last synced: 12 Nov 2024

https://github.com/Schlumberger/joint-vae

Pytorch implementation of JointVAE, a framework for disentangling continuous and discrete factors of variation :star2:

disentangled-representations disentanglement generative-models gumbel-softmax pytorch vae

Last synced: 14 Nov 2024

https://github.com/sinAshish/Multi-Scale-Attention

Code for our paper "Multi-scale Guided Attention for Medical Image Segmentation"

attention-mechanism biomedical-image-analysis deep-learning encoder-decoder pytorch

Last synced: 15 Nov 2024

https://github.com/yinchangchang/ocr_densenet

第一届西安交通大学人工智能实践大赛(2018AI实践大赛--图片文字识别)第一名;仅采用densenet识别图中文字

densenet ocr ocr-recognition python pytorch

Last synced: 03 Nov 2024

https://github.com/layumi/aicity-reid-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)

aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776

Last synced: 14 Nov 2024

https://github.com/layumi/AICIty-reID-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)

aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776

Last synced: 08 Nov 2024

https://github.com/farukalamai/advanced-machine-learning-engineer-roadmap-2024

A Full Stack ML (Machine Learning) Roadmap involves learning the necessary skills and technologies to become proficient in all aspects of machine learning, including data collection and preprocessing, model development, deployment, and maintenance.

aws computer-vision data-analysis data-science data-visualization deep-learning git-github machine-learning machine-learning-roadmap mlops natural-language-processing neural-network nlp opencv pandas python pytorch statistics tensorflow yolo

Last synced: 14 Nov 2024

https://github.com/dfdazac/wassdistance

Approximating Wasserstein distances with PyTorch

deep-learning machine-learning optimal-transport pytorch

Last synced: 12 Nov 2024

https://github.com/zalandoresearch/pytorch-vq-vae

PyTorch implementation of VQ-VAE by Aäron van den Oord et al.

deep-learning pytorch vae vq-vae

Last synced: 10 Nov 2024

https://github.com/walktree/libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm

cpp libtorch pytorch yolov3

Last synced: 09 Nov 2024

https://github.com/clovaai/rexnet

Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models

deep-learning image-classification imagenet object-detection pretrained-model pytorch rexnet transfer-learning

Last synced: 12 Nov 2024

https://github.com/huggingface/large_language_model_training_playbook

An open collection of implementation tips, tricks and resources for training large language models

cuda large-language-models llm nccl nlp performance python pytorch scalability troubleshooting

Last synced: 11 Nov 2024

https://github.com/thudm/graphmae

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

deep-learning graph-neural-networks pytorch self-supervised-learning

Last synced: 14 Nov 2024

https://github.com/cuiziteng/Illumination-Adaptive-Transformer

[BMVC 2022] You Only Need 90K Parameters to Adapt Light: A Light Weight Transformer for Image Enhancement and Exposure Correction. SOTA for low light enhancement, 0.004 seconds try this for pre-processing.

bmvc exposure-correction image-enhancement image-reconstruction image-restoration low-level-vision low-light-enhance low-light-image-enhancement pytorch transformer-architecture transformer-models

Last synced: 06 Nov 2024

https://github.com/ploomber/sklearn-evaluation

Machine learning model evaluation made easy: plots, tables, HTML reports, experiment tracking and Jupyter notebook analysis.

data-science deep-learning jupyter-notebook machine-learning pytorch scikit-learn sklearn tensorflow

Last synced: 13 Oct 2024

https://github.com/chaitjo/geometric-gnn-dojo

Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks.

e3nn equivariance equivariant-networks geometric-deep-learning geometric-graphs graph-isomorphism graph-neural-networks pytorch pytorch-geometric

Last synced: 15 Nov 2024

https://github.com/RaivoKoot/Video-Dataset-Loading-Pytorch

Generic PyTorch dataset implementation to load and augment VIDEOS for deep learning training loops.

action-recognition dataloader deep-learning machine-learning pytorch video-dataset videos

Last synced: 10 Nov 2024

https://github.com/loretoparisi/CapsNet

CapsNet (Capsules Net) in Geoffrey E Hinton paper "Dynamic Routing Between Capsules" - State Of the Art

capsnet capsule-networks chainer deeplearning keras lasagne machine-learning mnist mxnet pytorch tensorflow

Last synced: 02 Nov 2024

https://github.com/mv-lab/InstructIR

[ECCV 2024] InstructIR: High-Quality Image Restoration Following Human Instructions https://huggingface.co/spaces/marcosv/InstructIR

computer-vision deblurring deep-learning dehazing denoising image-enhancement image-restoration inverse-problems language-model low-light-image-enhancement multi-task multimodal neural-network photography prompt pytorch super-resolution

Last synced: 03 Nov 2024

https://github.com/mikoto10032/automaticweightedloss

Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning

auxiliary-tasks deep-learning multi-task multi-task-learning pytorch weigh-losses

Last synced: 11 Nov 2024

https://github.com/sacmehta/ESPNetv2

A light-weight, power efficient, and general purpose convolutional neural network

cnn convolutional-neural-networks deep-learning efficient imagenet lightweight machine-learning pytorch semantic-segmentation

Last synced: 31 Oct 2024

https://github.com/hkchengrex/mivos

[CVPR 2021] Modular Interactive Video Object Segmentation: Interaction-to-Mask, Propagation and Difference-Aware Fusion. Semi-supervised VOS as well!

computer-vision cvpr2021 deep-learning interactive-segmentation pytorch segmentation video-object-segmentation video-segmentation

Last synced: 10 Nov 2024

https://github.com/iBelieveCJM/Tricks-of-Semi-supervisedDeepLeanring-Pytorch

PseudoLabel 2013, VAT, PI model, Tempens, MeanTeacher, ICT, MixMatch, FixMatch

fixmatch ict meanteacher mixmatch pi pseudo-label pytorch semi-supervised-learning ssl tempens vat

Last synced: 15 Nov 2024

https://github.com/sthalles/pytorch-byol

PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning

byol deep-learning pytorch representation-learning self-supervised-learning simclr unsupervised-learning

Last synced: 06 Nov 2024

https://github.com/THUDM/GraphMAE

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

deep-learning graph-neural-networks pytorch self-supervised-learning

Last synced: 08 Aug 2024

https://github.com/pykale/pykale

Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for interdisciplinary research, part of the 🔥PyTorch ecosystem. ⭐ Star to support our work!

computer-vision data-science deep-learning domain-adaptation graph-analysis knowledge-aware-learning machine-learning medical-image-analysis meta-learning multimodal multimodal-learning python pytorch transfer-learning

Last synced: 11 Oct 2024

https://github.com/ruiminshen/yolo2-pytorch

PyTorch implementation of the YOLO (You Only Look Once) v2

caffe2 computer-vision deep-learning deep-neural-networks object-detection onnx python pytorch yolo2

Last synced: 04 Nov 2024

https://github.com/lemonhu/ner-bert-pytorch

PyTorch solution of named entity recognition task Using Google AI's pre-trained BERT model.

chinese-ner entity-extraction google-bert information-extraction msra named-entity-recognition ner pytorch transformer

Last synced: 16 Nov 2024

https://github.com/nschaetti/EchoTorch

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.

artificial-intelligence artificial-neural-networks echo-state-networks machine-learning machine-learning-algorithms machinelearning neural-networks python python-toolkit pytorch recurrent-networks recurrent-neural-networks reservoir-computing torch

Last synced: 03 Aug 2024

https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2

Ultra Fast Deep Lane Detection With Hybrid Anchor Driven Ordinal Classification (TPAMI 2022)

deep-learning lane-detection pytorch

Last synced: 11 Nov 2024

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

Model Predictive Path Integral (MPPI) with approximate dynamics implemented in pytorch

approximate-dynamics controls model-predictive-control mppi pytorch

Last synced: 16 Nov 2024

https://github.com/lemonhu/NER-BERT-pytorch

PyTorch solution of named entity recognition task Using Google AI's pre-trained BERT model.

chinese-ner entity-extraction google-bert information-extraction msra named-entity-recognition ner pytorch transformer

Last synced: 02 Nov 2024

https://github.com/likyoo/open-cd

A Change Detection Repo Standing on the Shoulders of Giants

bit change-detection changer deep-learning pytorch transformer

Last synced: 03 Aug 2024

https://github.com/tlkh/ai-lab

All-in-one AI container for rapid prototyping

cuda data-science deep-learning docker jupyter nvidia pytorch tensorflow

Last synced: 14 Nov 2024

https://github.com/daniilrobnikov/vits2

VITS2: Improving Quality and Efficiency of Single-Stage Text-to-Speech with Adversarial Learning and Architecture Design

deep-learning pytorch speech speech-synthesis text-to-speech tts vits2 voice-conversion

Last synced: 27 Oct 2024

https://github.com/aamini/evidential-deep-learning

Learn fast, scalable, and calibrated measures of uncertainty using neural networks!

confidence deep-learning evidence neural-network neurips-2020 pytorch tensorflow uncertainty

Last synced: 13 Nov 2024

https://github.com/krasserm/perceiver-io

A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training

deep-learning machine-learning perceiver perceiver-ar perceiver-io pytorch pytorch-lightning

Last synced: 12 Nov 2024

https://github.com/cccntu/minlora

minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.

fastai huggingface pytorch pytorch-implementation pytorch-lightning

Last synced: 09 Nov 2024

https://github.com/skhadem/3D-BoundingBox

PyTorch implementation for 3D Bounding Box Estimation Using Deep Learning and Geometry

3d computer-vision localization pytorch

Last synced: 28 Oct 2024

https://github.com/M-Nauta/TCDF

Temporal Causal Discovery Framework (PyTorch): discovering causal relationships between time series

causality cnn forecasting machine-learning neural-networks prediction python pytorch timeseries

Last synced: 15 Nov 2024

https://github.com/kubeflow/mpi-operator

Kubernetes Operator for MPI-based applications (distributed training, HPC, etc.)

apache-mxnet distributed-computing horovod kubeflow kubernetes mpi pytorch tensorflow

Last synced: 14 Oct 2024

https://github.com/ashishpatel26/Real-time-ML-Project

A curated list of applied machine learning and data science notebooks and libraries across different industries.

application deep-learning deeplearning dl keras machine-learning machine-learning-algorithms machinelearning ml ml-application project pytorch real-time real-time-data rl tensorflow theano

Last synced: 01 Nov 2024

https://github.com/blackhc/toma

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory

data-science gpu machine-learning python pytorch

Last synced: 16 Nov 2024

https://github.com/zihangJiang/TokenLabeling

Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"

imagenet lv-vit pytorch segmentation transformer vision

Last synced: 13 Nov 2024

https://github.com/BlackHC/toma

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory

data-science gpu machine-learning python pytorch

Last synced: 15 Nov 2024

https://github.com/songquanpeng/pytorch-template

To be the world's best PyTorch project template.

boilerplates gan gans pytorch pytorch-template

Last synced: 13 Nov 2024

https://github.com/shm007g/LLaMA-Cult-and-More

Large Language Models for All, 🦙 Cult and More, Stay in touch !

alpaca chatgpt deepspeed ggml gpt gpt4 gptq llama llm loralib pytorch tensorflow transformers vicuna

Last synced: 27 Oct 2024