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/bigmb/unet-segmentation-pytorch-nest-of-unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet

imagesegmentation python3 pytorch segmentation tensorvision torch torchvision unet

Last synced: 14 Oct 2024

https://github.com/harry24k/adversarial-attacks-pytorch

PyTorch implementation of adversarial attacks [torchattacks]

adversarial-attacks deep-learning pytorch

Last synced: 09 Oct 2024

https://github.com/rlworkgroup/garage

A toolkit for reproducible reinforcement learning research.

pytorch reproducibility rl-algorithms tensorflow

Last synced: 14 Oct 2024

https://github.com/nobody132/masr

中文语音识别; Mandarin Automatic Speech Recognition;

chinese-speech-recognition mandarin-chinese pytorch speech-recognition

Last synced: 14 Oct 2024

https://github.com/ncullen93/torchsample

Train AI models efficiently on medical images using any framework

deep-learning keras medical-imaging neuroimaging pytorch

Last synced: 28 Aug 2024

https://github.com/nitrain/nitrain

Train AI models efficiently on medical images using any framework

deep-learning keras medical-imaging neuroimaging pytorch

Last synced: 13 Oct 2024

https://github.com/microsoft/deepspeed-mii

MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.

deep-learning inference pytorch

Last synced: 07 Oct 2024

https://github.com/curt-park/rainbow-is-all-you-need

Rainbow is all you need! A step-by-step tutorial from DQN to Rainbow

colab-notebook dqn gym-environment nbviewer pytorch rainbow reinforcement-learning

Last synced: 14 Oct 2024

https://github.com/open-mmlab/mmrotate

OpenMMLab Rotated Object Detection Toolbox and Benchmark

detection openmmlab pytorch rotated-object

Last synced: 15 Oct 2024

https://github.com/adobe-research/custom-diffusion

Custom Diffusion: Multi-Concept Customization of Text-to-Image Diffusion (CVPR 2023)

computer-vision customization diffusion-models few-shot fine-tuning pytorch text-to-image-generation

Last synced: 14 Oct 2024

https://github.com/IgorSusmelj/pytorch-styleguide

An unofficial styleguide and best practices summary for PyTorch

best-practices pytorch styleguide

Last synced: 02 Aug 2024

https://github.com/waleedka/hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

deeplearning keras pytorch tensorboard tensorflow visualization

Last synced: 13 Oct 2024

https://github.com/Curt-Park/rainbow-is-all-you-need

Rainbow is all you need! A step-by-step tutorial from DQN to Rainbow

colab-notebook dqn gym-environment nbviewer pytorch rainbow reinforcement-learning

Last synced: 31 Oct 2024

https://github.com/haitongli/knowledge-distillation-pytorch

A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility

cifar10 computer-vision dark-knowledge deep-neural-networks knowledge-distillation model-compression pytorch

Last synced: 14 Oct 2024

https://github.com/uber/petastorm

Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet format. It supports ML frameworks such as Tensorflow, Pytorch, and PySpark and can be used from pure Python code.

deep-learning machine-learning parquet parquet-files pyarrow pyspark pytorch sysml tensorflow

Last synced: 15 Oct 2024

https://github.com/yxlllc/DDSP-SVC

Real-time end-to-end singing voice conversion system based on DDSP (Differentiable Digital Signal Processing)

pytorch

Last synced: 08 Nov 2024

https://github.com/csailvision/gandissect

Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/

deep-learning gan generative-adversarial-network image-manipulation interactive-visualizations interpretable-ml pytorch

Last synced: 15 Oct 2024

https://github.com/shunsukesaito/PIFu

This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"

3d-printing 3d-reconstruction arvr computer-graphics fashion geometry-processing human iccv2019 pifu pytorch

Last synced: 06 Nov 2024

https://github.com/CSAILVision/gandissect

Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/

deep-learning gan generative-adversarial-network image-manipulation interactive-visualizations interpretable-ml pytorch

Last synced: 02 Aug 2024

https://github.com/continualai/avalanche

Avalanche: an End-to-End Library for Continual Learning based on PyTorch.

benchmarks continual-learning continualai deep-learning evaluation framework library lifelong-learning metrics pytorch strategies training

Last synced: 15 Oct 2024

https://github.com/hkchengrex/XMem

[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model

computer-vision deep-learning eccv-2022 eccv2022 pytorch segmentation video-object-segmentation video-segmentation

Last synced: 09 Nov 2024

https://github.com/swz30/restormer

[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.

cvpr2022 defocus-deblurring efficient-transformers high-resolution image-deblurring image-deraining image-restoration low-level-vision motion-deblurring pytorch transformer

Last synced: 14 Oct 2024

https://github.com/swz30/Restormer

[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.

cvpr2022 defocus-deblurring efficient-transformers high-resolution image-deblurring image-deraining image-restoration low-level-vision motion-deblurring pytorch transformer

Last synced: 28 Oct 2024

https://github.com/davda54/sam

SAM: Sharpness-Aware Minimization (PyTorch)

optimizer pytorch sam sharpness-aware

Last synced: 09 Oct 2024

https://github.com/hkchengrex/xmem

[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model

computer-vision deep-learning eccv-2022 eccv2022 pytorch segmentation video-object-segmentation video-segmentation

Last synced: 15 Oct 2024

https://github.com/wkentaro/pytorch-fcn

PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)

computer-vision convolutional-networks deep-learning fcn fcn8s pytorch semantic-segmentation

Last synced: 25 Sep 2024

https://github.com/thu-coai/CDial-GPT

A Large-scale Chinese Short-Text Conversation Dataset and Chinese pre-training dialog models

dialogue gpt gpt-2 lccc pytorch text-generation

Last synced: 03 Aug 2024

https://github.com/huggingface/transfer-learning-conv-ai

🦄 State-of-the-Art Conversational AI with Transfer Learning

chatbots deep-learning dialog gpt gpt-2 neural-networks nlp pytorch transfer-learning

Last synced: 15 Oct 2024

https://github.com/Lightning-Universe/lightning-flash

Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains

classification deep-learning fiftyone icevision machine-learning object-detection open3d pytorch pytorch-lightning pytorch-video tabular-data tasks-flash torch-geometric

Last synced: 10 Nov 2024

https://github.com/PyTorchLightning/lightning-flash

Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains

classification deep-learning fiftyone icevision machine-learning object-detection open3d pytorch pytorch-lightning pytorch-video tabular-data tasks-flash torch-geometric

Last synced: 08 Nov 2024

https://github.com/gram-ai/capsule-networks

A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".

capsule-networks machine-learning mnist pytorch

Last synced: 29 Oct 2024

https://github.com/lightning-universe/lightning-flash

Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains

classification deep-learning fiftyone icevision machine-learning object-detection open3d pytorch pytorch-lightning pytorch-video tabular-data tasks-flash torch-geometric

Last synced: 26 Sep 2024

https://github.com/apachecn/.github

ApacheCN 开源组织:公告、介绍、成员、活动、交流方式

dl ml python pytorch solidity spark

Last synced: 12 Oct 2024

https://shunsukesaito.github.io/PIFu/

This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"

3d-printing 3d-reconstruction arvr computer-graphics fashion geometry-processing human iccv2019 pifu pytorch

Last synced: 03 Aug 2024

https://github.com/shunsukesaito/pifu

This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"

3d-printing 3d-reconstruction arvr computer-graphics fashion geometry-processing human iccv2019 pifu pytorch

Last synced: 14 Oct 2024

https://github.com/svip-lab/impersonator

PyTorch implementation of our ICCV 2019 paper: Liquid Warping GAN: A Unified Framework for Human Motion Imitation, Appearance Transfer and Novel View Synthesis

gan pose pytorch

Last synced: 14 Oct 2024

https://github.com/deepset-ai/farm

:house_with_garden: Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.

bert deep-learning germanbert language-models ner nlp nlp-framework nlp-library pretrained-models pytorch question-answering roberta transfer-learning xlnet-pytorch

Last synced: 15 Oct 2024

https://github.com/deepset-ai/FARM

:house_with_garden: Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.

bert deep-learning germanbert language-models ner nlp nlp-framework nlp-library pretrained-models pytorch question-answering roberta transfer-learning xlnet-pytorch

Last synced: 04 Nov 2024

https://github.com/ContinualAI/avalanche

Avalanche: an End-to-End Library for Continual Learning based on PyTorch.

benchmarks continual-learning continualai deep-learning evaluation framework library lifelong-learning metrics pytorch strategies training

Last synced: 02 Aug 2024

https://github.com/thudm/cogdl

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)

gnn-model graph-classification graph-embedding graph-neural-networks leaderboard link-prediction node-classification pytorch

Last synced: 15 Oct 2024

https://github.com/THUDM/CogView

Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".

pretrained-models pytorch text-to-image transformers

Last synced: 08 Nov 2024

https://github.com/THUDM/cogdl

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)

gnn-model graph-classification graph-embedding graph-neural-networks leaderboard link-prediction node-classification pytorch

Last synced: 07 Nov 2024

https://github.com/THUDM/CogDL

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)

gnn-model graph-classification graph-embedding graph-neural-networks leaderboard link-prediction node-classification pytorch

Last synced: 02 Aug 2024

https://github.com/wxinlong/solo

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.

instance-segmentation object-detection pytorch solo solov2

Last synced: 14 Oct 2024

https://github.com/thudm/cogview

Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".

pretrained-models pytorch text-to-image transformers

Last synced: 14 Oct 2024

https://github.com/Lightning-Universe/lightning-bolts

Toolbox of models, callbacks, and datasets for AI/ML researchers.

ai gan image-processing machine-learning natural-language-processing pytorch supervised-learning

Last synced: 06 Nov 2024

https://github.com/argusswift/yolov4-pytorch

This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO

attention cbam mobilenetv2 mobilenetv3 object-detection pytorch senet yolov4

Last synced: 14 Oct 2024

https://github.com/argusswift/YOLOv4-pytorch

This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO

attention cbam mobilenetv2 mobilenetv3 object-detection pytorch senet yolov4

Last synced: 09 Nov 2024

https://github.com/graph4ai/graph4nlp

Graph4nlp is the library for the easy use of Graph Neural Networks for NLP. Welcome to visit our DLG4NLP website (https://dlg4nlp.github.io/index.html) for various learning resources!

deep-learning graph-neural-networks machine-learning natural-language-processing nlp pytorch

Last synced: 15 Oct 2024

https://github.com/pytorch/tnt

A lightweight library for PyTorch training tools and utilities

deep-learning machine-learning neural-network python pytorch reinforcement-learning

Last synced: 07 Oct 2024

https://github.com/vahidk/effectivepytorch

PyTorch tutorials and best practices.

deep-learning ebook machine-learning neural-network pytorch

Last synced: 15 Oct 2024

https://github.com/vahidk/EffectivePyTorch

PyTorch tutorials and best practices.

deep-learning ebook machine-learning neural-network pytorch

Last synced: 27 Oct 2024

https://github.com/ymcui/Chinese-XLNet

Pre-Trained Chinese XLNet(中文XLNet预训练模型)

natural-language-processing nlp pytorch tensorflow xlnet

Last synced: 31 Oct 2024

https://github.com/ymcui/chinese-xlnet

Pre-Trained Chinese XLNet(中文XLNet预训练模型)

natural-language-processing nlp pytorch tensorflow xlnet

Last synced: 28 Oct 2024

https://github.com/lightning-universe/lightning-bolts

Toolbox of models, callbacks, and datasets for AI/ML researchers.

ai gan image-processing machine-learning natural-language-processing pytorch supervised-learning

Last synced: 15 Oct 2024

https://github.com/vita-group/transgan

[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang

gan pytorch transformer transformer-encoder transformer-models

Last synced: 14 Oct 2024

https://github.com/nvlabs/pwc-net

PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)

caffe computer-vision cvpr2018 deeplearning optical-flow pwc-net pytorch

Last synced: 15 Oct 2024

https://github.com/VITA-Group/TransGAN

[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang

gan pytorch transformer transformer-encoder transformer-models

Last synced: 03 Aug 2024

https://github.com/LoSealL/VideoSuperResolution

A collection of state-of-the-art video or single-image super-resolution architectures, reimplemented in tensorflow.

carn dbpn dncnn edsr frvsr ntire2019 pytorch rcan rdn srcnn srgan srmd super-resolution tensorflow vdsr vespcn vsr

Last synced: 28 Oct 2024

https://github.com/airaria/textbrewer

A PyTorch-based knowledge distillation toolkit for natural language processing

bert distillation knowledge nlp pytorch

Last synced: 09 Oct 2024

https://github.com/airaria/TextBrewer

A PyTorch-based knowledge distillation toolkit for natural language processing

bert distillation knowledge nlp pytorch

Last synced: 03 Nov 2024