PyTorch | Ecosyste.ms: Awesome

https://github.com/xiaoyufenfei/Efficient-Segmentation-Networks

Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, ESPNetv2, LEDNet, ESNet, FSSNet, CGNet, DABNet, Fast-SCNN, ContextNet, FPENet, etc.)

camvid cityscapes computer-vision driving-scene-understanding efficient-segmentation-networks image-segmentation lightweight-semantic-segmentation neural-networks pytorch real-time-semantic-segmentation scene-understanding segmentation semantic-segmentation semantic-segmentation-models

Last synced: 31 Oct 2024

https://github.com/DataXujing/YOLO-v5

:art: Pytorch YOLO v5 训练自己的数据集超详细教程！！！ :art: (提供PDF训练教程下载）

object-detection pytorch yolov5

Last synced: 09 Nov 2024

https://github.com/nyoki-mtl/pytorch-EverybodyDanceNow

Implementation of Everybody Dance Now by pytorch

everybody-dance-now pytorch

Last synced: 04 Nov 2024

https://github.com/vietnh1009/quickdraw

Implementation of Quickdraw - an online game developed by Google

cnn computer-vision cv2 deep-learning deep-neural-networks deeplearning image-classification neural-network neural-networks opencv opencv-python python python3 pytorch quickdraw quickdraw-dataset

Last synced: 12 Nov 2024

https://github.com/uvipen/quickdraw

Implementation of Quickdraw - an online game developed by Google

cnn computer-vision cv2 deep-learning deep-neural-networks deeplearning image-classification neural-network neural-networks opencv opencv-python python python3 pytorch quickdraw quickdraw-dataset

Last synced: 15 Oct 2024

https://github.com/sirius-ai/LPRNet_Pytorch

Pytorch Implementation For LPRNet, A High Performance And Lightweight License Plate Recognition Framework.

ctc-loss license-plate-recognition lprnet plate-detection plate-recognition pytorch

Last synced: 28 Oct 2024

https://github.com/Denys88/rl_games

RL implementations

deep-learning pytorch reinforcement-learning

Last synced: 03 Nov 2024

https://github.com/sberbank-ai-lab/LightAutoML

LAMA - automatic model creation framework

automated-machine-learning automl blackbox classification data-science ensembling feature-engineering gradient-boosting kaggle lama linear-model model-selection multiclass nlp parameter-tuning pipeline pytorch regression stacking whitebox

Last synced: 07 Aug 2024

https://github.com/nvidia/retinanet-examples

Fast and accurate object detection with end-to-end GPU optimization

deep-learning neural-network object-detection python pytorch retinanet tensorrt

Last synced: 29 Oct 2024

https://github.com/jankrepl/deepdow

Portfolio optimization with deep learning.

allocation convex-optimization deep-learning finance machine-learning markowitz portfolio-optimization pytorch stock-price-prediction timeseries trading wealth-management

Last synced: 28 Oct 2024

https://github.com/uvipen/QuickDraw

Implementation of Quickdraw - an online game developed by Google

cnn computer-vision cv2 deep-learning deep-neural-networks deeplearning image-classification neural-network neural-networks opencv opencv-python python python3 pytorch quickdraw quickdraw-dataset

Last synced: 07 Aug 2024

https://github.com/NVIDIA/retinanet-examples

Fast and accurate object detection with end-to-end GPU optimization

deep-learning neural-network object-detection python pytorch retinanet tensorrt

Last synced: 07 Aug 2024

https://github.com/dvgodoy/pytorchstepbystep

Official repository of my book: "Deep Learning with PyTorch Step-by-Step: A Beginner's Guide"

cnn-pytorch deep-learning python pytorch pytorch-tutorial rnn-pytorch

Last synced: 13 Nov 2024

https://github.com/BaguaSys/bagua

Bagua Speeds up PyTorch

bagua deep-learning distributed distributed-computing distributed-systems machine-learning pytorch rust-lang

Last synced: 30 Oct 2024

https://github.com/jacobgil/pytorch-pruning

PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference

deep-learning pruning pytorch

Last synced: 13 Nov 2024

https://github.com/baguasys/bagua

Bagua Speeds up PyTorch

bagua deep-learning distributed distributed-computing distributed-systems machine-learning pytorch rust-lang

Last synced: 12 Oct 2024

https://github.com/lonepatient/bert-multi-label-text-classification

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

albert bert fine-tuning multi-label-classification nlp pytorch pytorch-implmention text-classification transformers xlnet

Last synced: 13 Nov 2024

https://github.com/denys88/rl_games

RL implementations

deep-learning pytorch reinforcement-learning

Last synced: 14 Oct 2024

https://github.com/moemen95/pytorch-project-template

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models

Last synced: 12 Nov 2024

https://github.com/moemen95/PyTorch-Project-Template

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models

Last synced: 02 Nov 2024

https://github.com/easezyc/deep-transfer-learning

A collection of implementations of deep domain adaptation algorithms

deep-transfer-learning domain-adaptation pytorch transfer-learning

Last synced: 13 Nov 2024

https://github.com/moemen95/Pytorch-Project-Template

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models

Last synced: 03 Aug 2024

https://github.com/amusi/PyTorch-From-Zero-To-One

PyTorch从入门到精通

deep-learning pytorch pytorch-tutorials

Last synced: 28 Oct 2024

https://github.com/switchablenorms/Switchable-Normalization

Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10779

convolutional-neural-networks deeplearning imagenet normalization pytorch

Last synced: 13 Nov 2024

https://github.com/allegro/allRank

allRank is a framework for training learning-to-rank neural models based on PyTorch.

click-model deep-learning information-retrieval learning-to-rank machine-learning ndcg python pytorch ranking transformer

Last synced: 11 Nov 2024

https://github.com/allegro/allrank

allRank is a framework for training learning-to-rank neural models based on PyTorch.

click-model deep-learning information-retrieval learning-to-rank machine-learning ndcg python pytorch ranking transformer

Last synced: 13 Nov 2024

https://github.com/eric-mingjie/network-slimming

Network Slimming (Pytorch) (ICCV 2017)

channel-pruning convolutional-neural-networks deep-learning pytorch sparsity

Last synced: 08 Nov 2024

https://github.com/frotms/paddleocr2pytorch

PaddleOCR inference in PyTorch. Converted from [PaddleOCR](https://github.com/PaddlePaddle/PaddleOCR)

crnn db multilingual-models ocr paddleocr paddlepaddle pytorch pytorchocr

Last synced: 15 Oct 2024

https://github.com/dvgodoy/PyTorchStepByStep

Official repository of my book: "Deep Learning with PyTorch Step-by-Step: A Beginner's Guide"

cnn-pytorch deep-learning python pytorch pytorch-tutorial rnn-pytorch

Last synced: 28 Oct 2024

https://github.com/explosion/curated-transformers

🤖 A PyTorch library of curated Transformer models and their composable components

albert bert camembert dolly2 falcon gptneox llama llm llms nlp pytorch roberta transformer transformers xlm-roberta

Last synced: 07 Oct 2024

https://github.com/ypwhs/dl-engineer-guidebook

深度学习工程师生存指南

deep-learning deeplearning gitbook guidebook hardware homebrew installation-notes linux macos pytorch tensorflow

Last synced: 14 Nov 2024

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

Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.

alexnet cnn convolutional-networks convolutional-neural-network convolutional-neural-networks image-classification lenet pytorch pytorch-cnn pytorch-implementation pytorch-implmention pytorch-tutorial pytorch-tutorials resnet torchvision tutorial vgg

Last synced: 14 Nov 2024

https://github.com/opendilab/lightzero

[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios

alpha-beta-pruning alphazero atari board-game board-games continuous-control efficientzero gomoku gumbel-muzero gym mcts mcts-algorithm monte-carlo-tree-search muzero pytorch reinforcement-learning sampled-muzero self-play stochastic-muzero tictactoe

Last synced: 13 Oct 2024

https://github.com/gmalivenko/pytorch2keras

PyTorch to Keras model convertor

converter darts deep-learning deep-learning-models densenet imagenet keras keras-model-convertor keras-models keras-neural-networks keras-tensorflow machinelearning models-converter onnx pytorch pytorch-models resnet tensorflow tensorflow-models tensorflowjs

Last synced: 01 Nov 2024

https://github.com/philipperemy/n-beats

Keras/Pytorch implementation of N-BEATS: Neural basis expansion analysis for interpretable time series forecasting.

deep-learning neural-networks pytorch series-forecasting

Last synced: 21 Oct 2024

https://github.com/NirAharon/BoT-SORT

BoT-SORT: Robust Associations Multi-Pedestrian Tracking

multi-object-tracker multi-object-tracking multi-pedestrian-tracking pytorch tracking-by-detection yolov7 yolox

Last synced: 27 Oct 2024

https://github.com/ZhiqingXiao/rl-book

Source codes for the book "Reinforcement Learning: Theory and Python Implementation"

deep-reinforcement-learning gym openai-gym python pytorch reinforcement-learning tensorflow tensorflow2

Last synced: 03 Nov 2024

https://github.com/microsoft/godel

Large-scale pretrained models for goal-directed dialog

conversational-ai data-processing dialogpt dialogue dialogue-systems grounded-generation language-grounding language-model machine-learning pretrained-model pytorch text-data text-generation transformer transformers

Last synced: 14 Nov 2024

https://github.com/mlmed/torchxrayvision

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

chest-radiographs chest-xray chest-xray-images cxr cxr-images dataset deep-learning foundation-models image-classification machine-learning medical medical-ai medical-application medical-image-analysis medical-image-processing medical-imaging pytorch torchxrayvision transfer-learning

Last synced: 03 Aug 2024

https://github.com/lonePatient/Bert-Multi-Label-Text-Classification

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

albert bert fine-tuning multi-label-classification nlp pytorch pytorch-implmention text-classification transformers xlnet

Last synced: 04 Aug 2024

https://github.com/airctic/icevision

An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come

ai annotation-parsers coco-dataset coco-parser computer-vision deep-learning effecientdet fastai faster-rcnn mask-rcnn object-detection pycocotools python pytorch pytorch-lightning tutorials voc-dataset voc-parser

Last synced: 14 Nov 2024

https://github.com/jacobgil/vit-explain

Explainability for Vision Transformers

deep-learning explainable-ai pytorch transformer vision-transformer

Last synced: 13 Nov 2024

https://github.com/perone/medicaltorch

A medical imaging framework for Pytorch

computer-vision deep-learning machine-learning medical-imaging python pytorch

Last synced: 13 Nov 2024

https://github.com/liyuanlucasliu/lm-lstm-crf

Empower Sequence Labeling with Task-Aware Language Model

crf language-model ner pytorch sequence-labeling

Last synced: 09 Nov 2024

https://github.com/vahidk/tfrecord

Standalone TFRecord reader/writer with PyTorch data loaders

dataset loader pytorch tensorflow tfrecord

Last synced: 03 Aug 2024

https://github.com/microsoft/GODEL

Large-scale pretrained models for goal-directed dialog

conversational-ai data-processing dialogpt dialogue dialogue-systems grounded-generation language-grounding language-model machine-learning pretrained-model pytorch text-data text-generation transformer transformers

Last synced: 30 Oct 2024

https://github.com/zisianw/FaceBoxes.PyTorch

A PyTorch Implementation of FaceBoxes

face-detection faceboxes pytorch

Last synced: 28 Oct 2024

https://github.com/wasidennis/AdaptSegNet

Learning to Adapt Structured Output Space for Semantic Segmentation, CVPR 2018 (spotlight)

adversarial-learning computer-vision deep-learning domain-adaptation generative-adversarial-network pytorch semantic-segmentation

Last synced: 05 Aug 2024

https://github.com/progamergov/neural-style-pt

PyTorch implementation of neural style transfer algorithm

deep-learning deep-style neural-style neural-style-pt neural-style-transfer nst pytorch pytorch-style-transfer style-transfer styletransfer

Last synced: 13 Nov 2024

https://github.com/NVlabs/ODISE

Official PyTorch implementation of ODISE: Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models [CVPR 2023 Highlight]

deep-learning diffusion-models instance-segmentation open-vocabulary open-vocabulary-segmentation open-vocabulary-semantic-segmentation open-world-classification open-world-object-detection panoptic-segmentation pytorch semantic-segmentation text-image-retrieval zero-shot-learning

Last synced: 31 Oct 2024

https://github.com/minzhang-whu/change-detection-review

A review of change detection methods, including codes and open data sets for deep learning. From paper: change detection based on artificial intelligence: state-of-the-art and challenges.

artificial-intelligence caffe change-detection code deep-learning keras machine-learning matlab open-datasets python pytorch remote-sensing review sar streetview tensorflow unsupervised-learning

Last synced: 14 Nov 2024

https://github.com/MinZHANG-WHU/Change-Detection-Review

A review of change detection methods, including codes and open data sets for deep learning. From paper: change detection based on artificial intelligence: state-of-the-art and challenges.

artificial-intelligence caffe change-detection code deep-learning keras machine-learning matlab open-datasets python pytorch remote-sensing review sar streetview tensorflow unsupervised-learning

Last synced: 01 Nov 2024

https://github.com/RL-VIG/LibFewShot

LibFewShot: A Comprehensive Library for Few-shot Learning. TPAMI 2023.

few-shot-learning fine-tuning image-classification meta-learning pytorch

Last synced: 03 Aug 2024

https://github.com/baal-org/baal

Bayesian active learning library for research and industrial usecases.

active-learning ai bayesian-active-learning deep-learning machine-learning python pytorch

Last synced: 03 Aug 2024

https://github.com/gordicaleksa/pytorch-original-transformer

My implementation of the original transformer model (Vaswani et al.). I've additionally included the playground.py file for visualizing otherwise seemingly hard concepts. Currently included IWSLT pretrained models.

attention attention-is-all-you-need attention-mechanism deep-learning deeplearning jupyter original-transformer python pytorch pytorch-transformer pytorch-transformers transformer transformer-tutorial transformers

Last synced: 11 Nov 2024

https://github.com/bamos/densenet.pytorch

A PyTorch implementation of DenseNet.

deep-learning densenet pytorch

Last synced: 13 Nov 2024

https://github.com/hkchengrex/cascadepsp

[CVPR 2020] CascadePSP: Toward Class-Agnostic and Very High-Resolution Segmentation via Global and Local Refinement

computer-vision cvpr2020 deep-learning high-resolution pytorch refinement-network segmentation segmentation-refinement

Last synced: 10 Nov 2024

https://github.com/BUPT-GAMMA/OpenHGNN

This is an open-source toolkit for Heterogeneous Graph Neural Network(OpenHGNN) based on DGL.

dgl graph-neural-networks heterogeneous pytorch

Last synced: 08 Aug 2024

https://github.com/yzhangcs/parser

:rocket: State-of-the-art parsers for natural language.

constituency-parsing dependency-parsing pytorch semantic-dependency-parsing semiring state-of-the-art structured-prediction transformers

Last synced: 01 Oct 2024

https://github.com/voldemortX/pytorch-auto-drive

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help

apex cityscapes culane deeplab erfnet fcn gtav lane-detection lstr mobilenet onnx pascal-voc pytorch resa scnn semantic-segmentation tensorboard tensorrt tusimple

Last synced: 28 Oct 2024

https://github.com/braindotai/Watermark-Removal-Pytorch

🔥 CNN for Watermark Removal using Deep Image Prior with Pytorch 🔥.

artefacts-removal deep-image-priors deep-learning paper pytorch restoration-tasks skip-connections watermark watermark-removal

Last synced: 08 Nov 2024

https://github.com/redisai/redisai

A Redis module for serving tensors and executing deep learning graphs

machine-learning onnxruntime pytorch redisai serving-tensors tensorflow

Last synced: 14 Nov 2024

https://github.com/nyukat/breast_cancer_classifier

Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening

breast-cancer breast-cancer-diagnosis classification deep-learning medical-image-analysis medical-imaging neural-network pretrained-models pytorch tensorflow

Last synced: 04 Aug 2024

https://github.com/pytorch/translate

Translate - a PyTorch Language Library

artificial-intelligence machine-learning onnx pytorch

Last synced: 26 Sep 2024

https://github.com/smilelight/lightNLP

基于Pytorch和torchtext的自然语言处理深度学习框架。

cbow chinese cws deep-learning dependency-parsing hierarchical-softmax language-model nature-language-process negative-sampling nlp pos python pytorch relation-extraction sentence-similarity skip-gram srl text-classification torchtext word2vec

Last synced: 03 Aug 2024

https://github.com/RedisAI/RedisAI

A Redis module for serving tensors and executing deep learning graphs

machine-learning onnxruntime pytorch redisai serving-tensors tensorflow

Last synced: 30 Oct 2024

https://github.com/kakaobrain/torchgpipe

A GPipe implementation in PyTorch

checkpointing deep-learning gpipe model-parallelism parallelism pipeline-parallelism pytorch

Last synced: 10 Nov 2024

https://github.com/nashory/pggan-pytorch

:fire::fire: PyTorch implementation of "Progressive growing of GANs (PGGAN)" :fire::fire:

celeba-hq-dataset gan generative-adversarial-network progressive-gan progressively-growing-gan pytorch tensorboard

Last synced: 05 Nov 2024

https://github.com/diegoalejogm/gans

Generative Adversarial Networks implemented in PyTorch and Tensorflow

cyclegan dcgans gans gans-collections paper-implementations pytorch tensorflow

Last synced: 07 Aug 2024

https://github.com/havakv/pycox

Survival analysis with PyTorch

deep-learning machine-learning neural-networks python pytorch survival-analysis

Last synced: 14 Nov 2024

https://github.com/geoopt/geoopt

Riemannian Adaptive Optimization Methods with pytorch optim

optimization pytorch riemannian-geometry riemannian-manifold riemannian-optimization

Last synced: 30 Oct 2024

https://github.com/MontaEllis/Pytorch-Medical-Segmentation

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.

medical-image-processing pytorch segmentation torchio

Last synced: 07 Aug 2024

https://github.com/mimoralea/gdrl

Grokking Deep Reinforcement Learning

algorithms artificial-intelligence deep-learning deep-reinforcement-learning docker gpu machine-learning neural-networks numpy numpy-tutorial nvidia-docker pytorch pytorch-tutorials reinforcement-learning

Last synced: 13 Nov 2024

https://github.com/agrimgupta92/sgan

Code for "Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networks", Gupta et al, CVPR 2018

deep-learning generative-adversarial-network human-trajectory-prediction pytorch social-navigation trajectory-prediction

Last synced: 10 Nov 2024

https://github.com/cloneofsimo/minDiffusion

Self-contained, minimalistic implementation of diffusion models with Pytorch.

diffusion pytorch

Last synced: 30 Oct 2024

https://github.com/haofanwang/controlnet-for-diffusers

Transfer the ControlNet with any basemodel in diffusers🔥

aigc controlnet diffusers pytorch tutorials

Last synced: 14 Nov 2024

https://github.com/doragd/chinese-chatbot-pytorch-implementation

:four_leaf_clover: Another Chinese chatbot implemented in PyTorch, which is the sub-module of intelligent work order processing robot. 👩‍🔧

chatbot chinese-nlp deep-learning pytorch pytorch-nlp

Last synced: 29 Oct 2024

https://github.com/kimhc6028/relational-networks

Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)

deep-learning pytorch relational-networks

Last synced: 02 Nov 2024

https://github.com/wujiyang/Face_Pytorch

face recognition algorithms in pytorch framework, including arcface, cosface, sphereface and so on

arcface cosface face-recognition pytorch sphereface

Last synced: 28 Oct 2024

https://github.com/Replicable-MARL/MARLlib

One repository is all that is necessary for Multi-agent Reinforcement Learning (MARL)

deep-reinforcement-learning multi-agent-reinforcement-learning pytorch ray rllib

Last synced: 30 Oct 2024

https://github.com/mert-kurttutan/torchview

torchview: visualize pytorch models

deep-learning keras machine-learning neural-network python pytorch torchviz visualization

Last synced: 13 Oct 2024

https://github.com/datadreamer-dev/DataDreamer

DataDreamer: Prompt. Generate Synthetic Data. Train & Align Models. 🤖💤

alignment deep-learning fine-tuning gpt instruction-tuning llm llmops llms machine-learning natural-language-processing nlp nlp-library openai python pytorch synthetic-data synthetic-dataset-generation transformers

Last synced: 13 Oct 2024

https://github.com/bayesiains/nflows

Normalizing flows in PyTorch

density-estimation generative-model normalizing-flows pytorch

Last synced: 25 Oct 2024

https://github.com/alibaba/BladeDISC

BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.

compiler deep-learning inference-optimization machine-learning mlir neural-network pytorch tensorflow

Last synced: 27 Oct 2024

https://github.com/megvii-research/IJCAI2023-CoNR

IJCAI2023 - Collaborative Neural Rendering using Anime Character Sheets

anime computer-graphics computer-vision deep-learning pytorch

Last synced: 08 Nov 2024

https://github.com/denis2054/transformers-for-nlp-2nd-edition

Transformer models from BERT to GPT-4, environments from Hugging Face to OpenAI. Fine-tuning, training, and prompt engineering examples. A bonus section with ChatGPT, GPT-3.5-turbo, GPT-4, and DALL-E including jump starting GPT-4, speech-to-text, text-to-speech, text to image generation with DALL-E, Google Cloud AI,HuggingGPT, and more

bert chatgpt chatgpt-api dall-e dall-e-api deep-learning gpt-3-5-turbo gpt-4 gpt-4-api huggingface-transformers machine-learning natural-language-processing nlp openai python pytorch roberta-model transformers trax