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/xinntao/BasicSR

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.

basicsr basicvsr dfdnet ecbsr edsr edvr esrgan pytorch rcan restoration srgan srresnet stylegan2 super-resolution swinir

Last synced: 05 Aug 2024

https://github.com/XPixelGroup/BasicSR

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.

basicsr basicvsr dfdnet ecbsr edsr edvr esrgan pytorch rcan restoration srgan srresnet stylegan2 super-resolution swinir

Last synced: 31 Jul 2024

https://github.com/jessevig/bertviz

BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)

bert gpt2 machine-learning natural-language-processing neural-network nlp pytorch roberta transformer transformers visualization

Last synced: 24 Sep 2024

https://github.com/nadermx/backgroundremover

Background Remover lets you Remove Background from images and video using AI with a simple command line interface that is free and open source.

ai background-removal background-remover backgroundremover photo-editing python pytorch remove-background remove-background-image remove-background-video removebackground video video-editing

Last synced: 31 Jul 2024

https://github.com/jaywalnut310/vits

VITS: Conditional Variational Autoencoder with Adversarial Learning for End-to-End Text-to-Speech

deep-learning pytorch speech-synthesis text-to-speech tts

Last synced: 01 Aug 2024

https://github.com/codertimo/BERT-pytorch

Google AI 2018 BERT pytorch implementation

bert language-model nlp pytorch transformer

Last synced: 01 Aug 2024

https://github.com/KevinMusgrave/pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

computer-vision contrastive-learning deep-learning deep-metric-learning embeddings image-retrieval machine-learning metric-learning pytorch self-supervised-learning

Last synced: 01 Aug 2024

https://github.com/aymericdamien/topdeeplearning

A list of popular github projects related to deep learning

deep-learning machine-learning pytorch tensorflow

Last synced: 24 Sep 2024

https://github.com/kuangliu/pytorch-cifar

95.47% on CIFAR10 with PyTorch

pytorch

Last synced: 02 Aug 2024

https://github.com/aymericdamien/TopDeepLearning

A list of popular github projects related to deep learning

deep-learning machine-learning pytorch tensorflow

Last synced: 30 Jul 2024

https://github.com/microsoft/mmdnn

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.

caffe cntk coreml darknet keras model-converter mxnet onnx pytorch tensorflow visualization

Last synced: 24 Sep 2024

https://github.com/Microsoft/MMdnn

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.

caffe cntk coreml darknet keras model-converter mxnet onnx pytorch tensorflow visualization

Last synced: 01 Aug 2024

https://github.com/microsoft/MMdnn

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.

caffe cntk coreml darknet keras model-converter mxnet onnx pytorch tensorflow visualization

Last synced: 01 Aug 2024

https://github.com/nvidia/fastertransformer

Transformer related optimization, including BERT, GPT

bert gpt pytorch transformer

Last synced: 02 Aug 2024

https://github.com/skorch-dev/skorch

A scikit-learn compatible neural network library that wraps PyTorch

hacktoberfest huggingface machine-learning pytorch scikit-learn

Last synced: 24 Sep 2024

https://github.com/NVIDIA/FasterTransformer

Transformer related optimization, including BERT, GPT

bert gpt pytorch transformer

Last synced: 31 Jul 2024

https://github.com/modelscope/funasr

A Fundamental End-to-End Speech Recognition Toolkit and Open Source SOTA Pretrained Models, Supporting Speech Recognition, Voice Activity Detection, Text Post-processing etc.

audio-visual-speech-recognition conformer dfsmn paraformer pretrained-model punctuation pytorch rnnt speaker-diarization speech-recognition speechgpt speechllm vad voice-activity-detection whisper

Last synced: 24 Sep 2024

https://github.com/RangiLyu/nanodet

NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥

anchor-free android deep-learning deep-neural-networks efficientnet mnn model-zoo nanodet nanodet-plus ncnn object-detection openvino pytorch repvgg shufflenet

Last synced: 30 Jul 2024

https://github.com/meituan/YOLOv6

YOLOv6: a single-stage object detection framework dedicated to industrial applications.

object-detection pytorch yolo

Last synced: 30 Jul 2024

https://github.com/facebookresearch/mmf

A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)

captioning deep-learning dialog hateful-memes multi-tasking multimodal pretrained-models pytorch textvqa vqa

Last synced: 01 Aug 2024

https://github.com/vwxyzjn/cleanrl

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

a2c actor-critic advantage-actor-critic ale atari deep-learning deep-reinforcement-learning gym machine-learning phasic-policy-gradient ppo proximal-policy-optimization python pytorch reinforcement-learning wandb

Last synced: 24 Sep 2024

https://github.com/udacity/deep-learning-v2-pytorch

Projects and exercises for the latest Deep Learning ND program https://www.udacity.com/course/deep-learning-nanodegree--nd101

convolutional-networks deep-learning neural-network pytorch recurrent-networks sentiment-analysis style-transfer

Last synced: 01 Aug 2024

https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.

bifpn detection efficientdet efficientnet object-detection pytorch

Last synced: 31 Jul 2024

https://github.com/google/gemma_pytorch

The official PyTorch implementation of Google's Gemma models

gemma google pytorch

Last synced: 31 Jul 2024

https://github.com/yunjey/stargan

StarGAN - Official PyTorch Implementation (CVPR 2018)

cvpr2018 generative-models image-to-image-translation pytorch stargan

Last synced: 01 Aug 2024

https://github.com/pyannote/pyannote-audio

Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech detection, speaker embedding

overlapped-speech-detection pretrained-models pytorch speaker-change-detection speaker-diarization speaker-embedding speaker-recognition speaker-verification speech-activity-detection speech-processing voice-activity-detection

Last synced: 31 Jul 2024

https://github.com/timeseriesai/tsai

Time series Timeseries Deep Learning Machine Learning Python Pytorch fastai | State-of-the-art Deep Learning library for Time Series and Sequences in Pytorch / fastai

classification cnn deep-learning fastai forecasting inceptiontime machine-learning python pytorch regression rnn rocket self-supervised sequential state-of-the-art time-series time-series-analysis time-series-classification timeseries transformer

Last synced: 24 Sep 2024

https://github.com/dbolya/yolact

A simple, fully convolutional model for real-time instance segmentation.

instance-segmentation pytorch real-time realtime yolact

Last synced: 24 Sep 2024

https://github.com/nvidia/dali

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

audio-processing data-augmentation data-processing deep-learning fast-data-pipeline gpu gpu-tensorflow image-augmentation image-processing machine-learning mxnet neural-network paddle python pytorch

Last synced: 24 Sep 2024

https://github.com/NVIDIA/DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

audio-processing data-augmentation data-processing deep-learning fast-data-pipeline gpu gpu-tensorflow image-augmentation image-processing machine-learning mxnet neural-network paddle python pytorch

Last synced: 30 Jul 2024

https://github.com/zhouhaoyi/Informer2020

The GitHub repository for the paper "Informer" accepted by AAAI 2021.

deep-learning forecasting pytorch self-attention time-series transformer

Last synced: 01 Aug 2024

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

OpenMMLab's next-generation platform for general 3D object detection.

3d-object-detection object-detection point-cloud pytorch

Last synced: 31 Jul 2024

https://github.com/timeseriesAI/tsai

Time series Timeseries Deep Learning Machine Learning Python Pytorch fastai | State-of-the-art Deep Learning library for Time Series and Sequences in Pytorch / fastai

classification cnn deep-learning fastai forecasting inceptiontime machine-learning python pytorch regression rnn rocket self-supervised sequential state-of-the-art time-series time-series-analysis time-series-classification timeseries transformer

Last synced: 31 Jul 2024

https://github.com/roboflow/notebooks

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models like Grounding DINO and SAM.

amazon-sagemaker-lab automatic-labeling-system computer-vision deep-learning deep-neural-networks google-colab image-classification image-segmentation machine-learning object-detection open-vocabulary-detection open-vocabulary-segmentation pytorch tutorial yolov5 yolov6 yolov7 yolov8 zero-shot-classification zero-shot-detection

Last synced: 31 Jul 2024

https://github.com/xorbitsai/inference

Replace OpenAI GPT with another LLM in your app by changing a single line of code. Xinference gives you the freedom to use any LLM you need. With Xinference, you're empowered to run inference with any open-source language models, speech recognition models, and multimodal models, whether in the cloud, on-premises, or even on your laptop.

artificial-intelligence chatglm deployment flan-t5 gemma ggml glm4 inference llama llama3 llamacpp llm machine-learning mistral openai-api pytorch qwen vllm whisper wizardlm

Last synced: 24 Sep 2024

https://github.com/lyhue1991/eat_pytorch_in_20_days

Pytorch🍊🍉 is delicious, just eat it! 😋😋

deep-learning pytorch

Last synced: 01 Aug 2024

https://github.com/CSAILVision/semantic-segmentation-pytorch

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset

ade20k pytorch scene-recognition semantic-segmentation

Last synced: 05 Aug 2024

https://github.com/snakers4/silero-models

Silero Models: pre-trained speech-to-text, text-to-speech and text-enhancement models made embarrassingly simple

asr capitalization colab english german onnx pretrained-models pytorch repunctuation spanish speech speech-recognition speech-synthesis speech-to-text stt stt-benchmark text-to-speech torch-hub tts tts-models

Last synced: 31 Jul 2024

https://github.com/deep-learning-with-pytorch/dlwpt-code

Code for the book Deep Learning with PyTorch by Eli Stevens, Luca Antiga, and Thomas Viehmann.

deep-learning deep-neural-networks python python3 pytorch

Last synced: 31 Jul 2024

https://github.com/OAID/Tengine

Tengine is a lite, high performance, modular inference engine for embedded device

acl arm artificial-intelligence cnn container cuda machine-learning mips npu nvdla onnx pytorch riscv supperedge tensorflow tensorrt x86-64

Last synced: 31 Jul 2024

https://github.com/superduper-io/superduper

🔮 SuperDuper: Bring AI to your database! Build, deploy and manage any AI application directly with your existing data infrastructure, without moving your data. Including streaming inference, scalable model training and vector search.

ai chatbot data database distributed-ml inference llm-inference llm-serving llmops ml mlops mongodb pretrained-models python pytorch rag semantic-search torch transformers vector-search

Last synced: 24 Sep 2024

https://github.com/pytorch/ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

closember deep-learning hacktoberfest machine-learning metrics neural-network python pytorch

Last synced: 24 Sep 2024

https://github.com/NVIDIA-AI-IOT/torch2trt

An easy to use PyTorch to TensorRT converter

classification inference jetson-nano jetson-tx2 jetson-xavier pytorch tensorrt

Last synced: 31 Jul 2024

https://github.com/hunkim/deeplearningzerotoall

TensorFlow Basic Tutorial Labs

keras lab mxnet python pytorch tensorflow tutorial

Last synced: 24 Sep 2024

https://github.com/hunkim/DeepLearningZeroToAll

TensorFlow Basic Tutorial Labs

keras lab mxnet python pytorch tensorflow tutorial

Last synced: 01 Aug 2024

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

PyTorch ,ONNX and TensorRT implementation of YOLOv4

darknet2onnx darknet2pytorch onnx pytorch pytorch-yolov4 tensorrt yolov3 yolov4 yolov4-tiny

Last synced: 31 Jul 2024

https://github.com/BoltzmannEntropy/interviews.ai

It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; however, it is my hope that even the most experienced researchers will find it fascinating as well.

artificial-intelligence autograd bayesian-statistics convolutional-neural-networks data-science deep-learning ensemble-learning feature-extraction graduate-school information-theory interview-preparation jax jobs logistic-regression loss-functions machine-learning python pytorch pytorch-tutorial

Last synced: 31 Jul 2024

https://github.com/sshaoshuai/PCDet

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.

3d-detection autonomous-driving object-detection point-cloud pv-rcnn pytorch

Last synced: 13 Aug 2024

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

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.

3d-detection autonomous-driving object-detection point-cloud pv-rcnn pytorch

Last synced: 31 Jul 2024

https://github.com/carperai/trlx

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

machine-learning pytorch reinforcement-learning

Last synced: 01 Aug 2024

https://github.com/CarperAI/trlx

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

machine-learning pytorch reinforcement-learning

Last synced: 30 Jul 2024

https://github.com/neuralchen/SimSwap

An arbitrary face-swapping framework on images and videos with one single trained model!

deepfacelab deepfakes face faceswap gan image-manipulation pytorch swap video

Last synced: 01 Aug 2024

https://github.com/apple/coremltools

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

coreml coremltools machine-learning model-conversion model-converter pytorch tensorflow

Last synced: 24 Sep 2024

https://github.com/ifzhang/ByteTrack

[ECCV 2022] ByteTrack: Multi-Object Tracking by Associating Every Detection Box

deployment multi-object-tracking pytorch real-time

Last synced: 30 Jul 2024

https://intellabs.github.io/distiller/

Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller

automl-for-compression deep-neural-networks distillation early-exit group-lasso jupyter-notebook network-compression onnx pruning pruning-structures pytorch quantization regularization truncated-svd

Last synced: 02 Aug 2024

https://github.com/IntelLabs/distiller

Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller

automl-for-compression deep-neural-networks distillation early-exit group-lasso jupyter-notebook network-compression onnx pruning pruning-structures pytorch quantization regularization truncated-svd

Last synced: 31 Jul 2024

https://nervanasystems.github.io/distiller/

Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller

automl-for-compression deep-neural-networks distillation early-exit group-lasso jupyter-notebook network-compression onnx pruning pruning-structures pytorch quantization regularization truncated-svd

Last synced: 04 Aug 2024

https://github.com/autogptq/autogptq

An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algorithm.

deep-learning inference large-language-models llms nlp pytorch quantization transformer transformers

Last synced: 24 Sep 2024

https://github.com/timesler/facenet-pytorch

Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models

face-detection face-identification face-recognition face-tracking facial-recognition inception-resnet mtcnn pretrained-models pytorch resnet

Last synced: 01 Aug 2024

https://github.com/Tencent/TNN

TNN: developed by Tencent Youtu Lab and Guangying Lab, a uniform deep learning inference framework for mobile、desktop and server. TNN is distinguished by several outstanding features, including its cross-platform capability, high performance, model compression and code pruning. Based on ncnn and Rapidnet, TNN further strengthens the support and performance optimization for mobile devices, and also draws on the advantages of good extensibility and high performance from existed open source efforts. TNN has been deployed in multiple Apps from Tencent, such as Mobile QQ, Weishi, Pitu, etc. Contributions are welcome to work in collaborative with us and make TNN a better framework.

coreml deep-learning face-detection hairsegmentaion inference mnn ncnn ocr openvino pytorch tengine tensorflow tensorrt

Last synced: 31 Jul 2024

https://github.com/dsgiitr/d2l-pytorch

This project reproduces the book Dive Into Deep Learning (https://d2l.ai/), adapting the code from MXNet into PyTorch.

book computer-vision d2l data-science deep-learning dive-into-deep-learning mxnet nlp pytorch pytorch-implmention

Last synced: 31 Jul 2024

https://github.com/pytorch/serve

Serve, optimize and scale PyTorch models in production

cpu deep-learning docker gpu kubernetes machine-learning metrics mlops optimization pytorch serving

Last synced: 24 Sep 2024

https://github.com/LaurentMazare/tch-rs

Rust bindings for the C++ api of PyTorch.

deep-learning machine-learning neural-network pytorch rust

Last synced: 31 Jul 2024

https://github.com/Fanghua-Yu/SUPIR

SUPIR aims at developing Practical Algorithms for Photo-Realistic Image Restoration In the Wild. Our new online demo is also released at suppixel.ai.

deep-learning diffusion-models llava pytorch pytorch-lightning restoration sdxl stable-diffusion super-resolution

Last synced: 31 Jul 2024

https://github.com/layumi/Person_reID_baseline_pytorch

:bouncing_ball_person: Pytorch ReID: A tiny, friendly, strong pytorch implement of person re-id / vehicle re-id baseline. Tutorial 👉https://github.com/layumi/Person_reID_baseline_pytorch/tree/master/tutorial

awesome-reid baseline circle-loss cuhk-np image-retrieval image-search market-1501 metric-learning msmt17 object-reid open-reid person-reid person-reidentification pytorch random-erasing re-ranking tutorial vehicle-reid

Last synced: 03 Aug 2024

https://github.com/weiaicunzai/pytorch-cifar100

Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet, NasNet, Residual Attention Network, SENet, WideResNet)

cifar100 deep-learning densenet googlenet image-classification inception-resnet-v2 inceptionv3 inceptionv4 mobilenet nasnet pytorch resnet resnext shufflenet squeezenet wideresnet xception

Last synced: 31 Jul 2024

https://github.com/sksq96/pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

deep-learning keras pytorch summary

Last synced: 24 Sep 2024

https://github.com/idea-ccnl/fengshenbang-lm

Fengshenbang-LM(封神榜大模型)是IDEA研究院认知计算与自然语言研究中心主导的大模型开源体系,成为中文AIGC和认知智能的基础设施。

aigc chinese-nlp distributed-training multimodal pretrained-models pytorch transformers

Last synced: 24 Sep 2024

https://github.com/nianticlabs/monodepth2

[ICCV 2019] Monocular depth estimation from a single image

computer-vision deep-learning depth-estimation monodepth neural-network pytorch self-supervision

Last synced: 03 Aug 2024

https://github.com/pqpo/SmartCropper

🔥 A library for cropping image in a smart way that can identify the border and correct the cropped image. 智能图片裁剪框架。自动识别边框,手动调节选区,使用透视变换裁剪并矫正选区;适用于身份证,名片,文档等照片的裁剪。

detection hed-net image image-cropper opencv pytorch smartcropper tensorflow

Last synced: 02 Aug 2024

https://github.com/wenet-e2e/wenet

Production First and Production Ready End-to-End Speech Recognition Toolkit

asr automatic-speech-recognition conformer e2e-models production-ready pytorch speech-recognition transformer whisper

Last synced: 01 Aug 2024