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/fendouai/PyTorchDocs

PyTorch 官方中文教程包含 60 分钟快速入门教程，强化教程，计算机视觉，自然语言处理，生成对抗网络，强化学习。欢迎 Star，Fork！

pytorch pytorch-cnn pytorch-gan pytorch-nlp pytorch-tutorial

Last synced: 03 Aug 2024

https://github.com/pytorch/TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

cuda deep-learning jetson libtorch machine-learning nvidia pytorch tensorrt

Last synced: 01 Aug 2024

https://github.com/mravanelli/pytorch-kaldi

pytorch-kaldi is a project for developing state-of-the-art DNN/RNN hybrid speech recognition systems. The DNN part is managed by pytorch, while feature extraction, label computation, and decoding are performed with the kaldi toolkit.

asr deep-learning deep-neural-networks dnn dnn-hmm gru kaldi lstm lstm-neural-networks multilayer-perceptron-network pytorch recurrent-neural-networks rnn rnn-model speech speech-recognition timit

Last synced: 30 Jul 2024

https://github.com/OAID/TengineKit

TengineKit - Free, Fast, Easy, Real-Time Face Detection & Face Landmarks & Face Attributes & Hand Detection & Hand Landmarks & Body Detection & Body Landmarks & Iris Landmarks & Yolov5 SDK On Mobile.

ai android artificial-intelligence computer-vision deep-neural-networks face-alignment face-api face-attributes face-detection face-landmarks face-tracking facial-landmarks java mobile pytorch tensorflow

Last synced: 02 Aug 2024

https://github.com/gordicaleksa/pytorch-GAT

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

attention attention-mechanism deep-learning gat gat-tutorial graph-attention-network graph-attention-networks jupyter python pytorch pytorch-gat pytorch-implementation self-attention

Last synced: 02 Aug 2024

https://github.com/tencentmusic/cube-studio

cube studio开源云原生一站式机器学习/深度学习AI平台，支持sso登录，多租户/多项目组，大数据平台对接，notebook在线开发，拖拉拽任务流pipeline编排，多机多卡分布式训练，超参搜索，推理服务VGPU，边缘计算，serverless，标注平台，自动化标注，数据集管理，大模型微调，vllm大模型推理，llmops，私有知识库，AI模型应用商店，支持模型一键开发/推理/微调，支持国产cpu/gpu/npu芯片，支持RDMA，支持pytorch/tf/mxnet/deepspeed/paddle/colossalai/horovod/spark/ray/volcano分布式

ai aihub argo automl gpt inference kubeflow kubernetes llmops mlops notebook pipeline pytorch spark vgpu workflow

Last synced: 30 Jul 2024

https://github.com/phlippe/uvadlc_notebooks

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

deep-learning flax jax optax pytorch pytorch-lightning tutorial tutorials uvadlc

Last synced: 09 Aug 2024

https://github.com/automl/Auto-PyTorch

Automatic architecture search and hyperparameter optimization for PyTorch

automl deep-learning pytorch tabular-data

Last synced: 01 Aug 2024

https://github.com/lakesoul-io/LakeSoul

LakeSoul is an end-to-end, realtime and cloud native Lakehouse framework with fast data ingestion, concurrent update and incremental data analytics on cloud storages for both BI and AI applications.

arrow big-data datafusion datalake flink huggingface lakehouse lakesoul postgresql python pytorch rust spark sql streaming vectorized velox

Last synced: 31 Jul 2024

https://github.com/curiousily/Getting-Things-Done-with-Pytorch

Jupyter Notebook tutorials on solving real-world problems with Machine Learning & Deep Learning using PyTorch. Topics: Face detection with Detectron 2, Time Series anomaly detection with LSTM Autoencoders, Object Detection with YOLO v5, Build your first Neural Network, Time Series forecasting for Coronavirus daily cases, Sentiment Analysis with BERT.

anomaly-detection bert computer-vision coronavirus deep-learning face-detection face-recognition lstm machine-learning nlp object-detection pytorch sentiment-analysis time-series time-series-anomaly-detection time-series-forecasting transfer-learning transformer tutorial yolo

Last synced: 01 Aug 2024

https://github.com/moskomule/senet.pytorch

PyTorch implementation of SENet

pytorch senet

Last synced: 31 Jul 2024

https://github.com/hzwer/ICCV2019-LearningToPaint

ICCV2019 - Learning to Paint With Model-based Deep Reinforcement Learning

computer-vision deep-learning painting pytorch reinforcement-learning

Last synced: 06 Aug 2024

https://github.com/deeplearningturkiye/turkce-yapay-zeka-kaynaklari

Türkiye'de yapılan derin öğrenme (deep learning) ve makine öğrenmesi (machine learning) çalışmalarının derlendiği sayfa.

caffe convolutional-neural-networks deep-learning derin-ogrenme dogal-dil-isleme keras machine-learning makine-ogrenmesi natural-language-processing pytorch tensorflow yapay-zeka

Last synced: 01 Aug 2024

https://github.com/microsoft/torchgeo

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

computer-vision datasets deep-learning earth-observation geospatial models pytorch remote-sensing satellite-imagery torchvision transforms

Last synced: 01 Aug 2024

https://github.com/LinkSoul-AI/Chinese-Llama-2-7b

开源社区第一个能下载、能运行的中文 LLaMA2 模型！

deep-learning llama2 llama2-docker llm pytorch

Last synced: 30 Jul 2024

https://github.com/state-spaces/s4

Structured state space sequence models

pytorch sequence-models state-space-models

Last synced: 31 Jul 2024

https://github.com/google-research-datasets/Objectron

Objectron is a dataset of short, object-centric video clips. In addition, the videos also contain AR session metadata including camera poses, sparse point-clouds and planes. In each video, the camera moves around and above the object and captures it from different views. Each object is annotated with a 3D bounding box. The 3D bounding box describes the object’s position, orientation, and dimensions. The dataset contains about 15K annotated video clips and 4M annotated images in the following categories: bikes, books, bottles, cameras, cereal boxes, chairs, cups, laptops, and shoes

3d 3d-reconstruction 3d-vision ai augmented-reality computer-vision dataset deep-learning machine-learning neural-network python pytorch tensorflow

Last synced: 31 Jul 2024

https://github.com/ufoym/imbalanced-dataset-sampler

A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.

data-sampling image-classification imbalanced-data pytorch

Last synced: 03 Aug 2024

https://github.com/PennyLaneAI/pennylane

PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.

autograd automatic-differentiation cirq deep-learning differentiable-computing hacktoberfest jax machine-learning neural-network optimization python pytorch qiskit qml quantum quantum-chemistry quantum-computing quantum-machine-learning strawberryfields tensorflow

Last synced: 31 Jul 2024

https://github.com/666DZY666/micronet

micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference)、Low-Bit(≤2b)/Ternary and Binary(TWN/BNN/XNOR-Net); post-training-quantization(PTQ), 8-bit(tensorrt); 2、 pruning: normal、regular and group convolutional channel pruning; 3、 group convolution structure; 4、batch-normalization fuse for quantization. deploy: tensorrt, fp32/fp16/int8(ptq-calibration)、op-adapt(upsample)、dynamic_shape

batch-normalization-fuse bnn convolutional-networks dorefa group-convolution integer-arithmetic-only model-compression network-in-network network-slimming neuromorphic-computing onnx post-training-quantization pruning pytorch quantization quantization-aware-training tensorrt tensorrt-int8-python twn xnor-net

Last synced: 31 Jul 2024

https://github.com/datawhalechina/thorough-pytorch

PyTorch入门教程，在线阅读地址：https://datawhalechina.github.io/thorough-pytorch/

deep-learning machine-learning python pytorch

Last synced: 03 Aug 2024

https://github.com/asteroid-team/asteroid

The PyTorch-based audio source separation toolkit for researchers

audio-separation deep-learning pretrained-models pytorch source-separation speech-enhancement speech-separation

Last synced: 05 Aug 2024

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

[ICRA'23] BEVFusion: Multi-Task Multi-Sensor Fusion with Unified Bird's-Eye View Representation

3d-perception camera lidar object-detection pytorch semantic-segmentation sensor-fusion

Last synced: 31 Jul 2024

https://github.com/PeterH0323/Smart_Construction

Base on YOLOv5 Head Person Helmet Detection on Construction Sites，基于目标检测工地安全帽和禁入危险区域识别系统，🚀😆附 YOLOv5 训练自己的数据集超详细教程🚀😆2021.3新增可视化界面❗❗

detection helmet object-detection python pytorch yolo yolov5

Last synced: 02 Aug 2024

https://github.com/yhenon/pytorch-retinanet

Pytorch implementation of RetinaNet object detection.

python pytorch retinanet

Last synced: 31 Jul 2024

https://github.com/CoinCheung/pytorch-loss

label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful

amsoftmax cuda dice-loss ema focal-loss label-smoothing lovasz-softmax mish partial-fc pytorch triplet-loss

Last synced: 03 Aug 2024

https://github.com/asappresearch/sru

Training RNNs as Fast as CNNs (https://arxiv.org/abs/1709.02755)

deep-learning nlp pytorch recurrent-neural-networks

Last synced: 06 Aug 2024

https://github.com/huggingface/optimum

🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools

graphcore habana inference intel onnx onnxruntime optimization pytorch quantization tflite training transformers

Last synced: 31 Jul 2024

https://github.com/ppogg/YOLOv5-Lite

🍅🍅🍅YOLOv5-Lite: Evolved from yolov5 and the size of model is only 900+kb (int8) and 1.7M (fp16). Reach 15 FPS on the Raspberry Pi 4B~

android-app mnn mobilenet ncnn onnxruntime openvivo picodet pplcnet pytorch repvgg shufflenetv2 tensorrt tflite transformer yolov5

Last synced: 31 Jul 2024

https://github.com/AIZOOTech/FaceMaskDetection

开源人脸口罩检测模型和数据 Detect faces and determine whether people are wearing mask.

caffe detection pytorch

Last synced: 31 Jul 2024

https://github.com/NVlabs/nvdiffrec

Official code for the CVPR 2022 (oral) paper "Extracting Triangular 3D Models, Materials, and Lighting From Images".

deep-learning pytorch

Last synced: 03 Aug 2024

https://nvlabs.github.io/nvdiffrec/

Official code for the CVPR 2022 (oral) paper "Extracting Triangular 3D Models, Materials, and Lighting From Images".

deep-learning pytorch

Last synced: 01 Aug 2024

https://github.com/aws/sagemaker-python-sdk

A library for training and deploying machine learning models on Amazon SageMaker

aws huggingface machine-learning mxnet python pytorch sagemaker tensorflow

Last synced: 02 Aug 2024

https://github.com/TingFree/NLPer-Arsenal

收录NLP竞赛策略实现、各任务baseline、相关竞赛经验贴（当前赛事、往期赛事、训练赛）、NLP会议时间、常用自媒体、GPU推荐等，持续更新中

baselines gpu nlp nlp-competition nlp-conference nlp-media pytorch

Last synced: 01 Aug 2024

https://github.com/Jack-Cherish/Deep-Learning

:computer:深度学习实战：手写数字识别、Discuz验证码识别、垃圾分类、语义分割

deep-learning python python3 pytorch tensorflow

Last synced: 31 Jul 2024

https://github.com/lonePatient/BERT-NER-Pytorch

Chinese NER(Named Entity Recognition) using BERT(Softmax, CRF, Span)

adversarial-training albert bert chinese crf focal-loss labelsmoothing ner nlp pytorch softmax span

Last synced: 03 Aug 2024

https://github.com/webdataset/webdataset

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

data-augmentation deep-learning pytorch webdataset webdataset-format

Last synced: 03 Aug 2024

https://github.com/Lightning-AI/torchmetrics

Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.

analyses data-science deep-learning machine-learning metrics python pytorch

Last synced: 31 Jul 2024

https://github.com/megvii-research/NAFNet

The state-of-the-art image restoration model without nonlinear activation functions.

deblur denoise eccv2022 image-deblurring image-denoising image-restoration low-level-vision pytorch stereo-super-resolution

Last synced: 07 Aug 2024

https://github.com/azavea/raster-vision

An open source library and framework for deep learning on satellite and aerial imagery.

classification computer-vision deep-learning geospatial machine-learning object-detection pytorch remote-sensing semantic-segmentation

Last synced: 31 Jul 2024

https://github.com/dmMaze/BallonsTranslator

深度学习辅助漫画翻译工具, 支持一键机翻和简单的图像/文本编辑 | Yet another computer-aided comic/manga translation tool powered by deeplearning

anime auto-translation chinese-translation comics computer-aided-translation computer-vision deep-learning inpainting manga ocr pyqt pyqt6 pytorch qt qt6 scene-text-detection

Last synced: 01 Aug 2024

https://github.com/WongKinYiu/yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)

darknet explicit implicit pytorch representation scaled-yolov4 unified-network yolo yolor yolov4 yolov4-csp

Last synced: 30 Jul 2024

https://github.com/HypoX64/DeepMosaics

Automatically remove the mosaics in images and videos, or add mosaics to them.

computer-vision deep-learning mosaic pytorch video-inpainting

Last synced: 01 Aug 2024

https://github.com/delip/PyTorchNLPBook

Code and data accompanying Natural Language Processing with PyTorch published by O'Reilly Media https://amzn.to/3JUgR2L

deep-learning deep-neural-networks natural-language-processing neural-machine-translation neural-networks nlp pytorch pytorch-nlp pytorch-tutorial

Last synced: 03 Sep 2024

https://github.com/salesforce/awd-lstm-lm

LSTM and QRNN Language Model Toolkit for PyTorch

language-model lstm pytorch qrnn sgd

Last synced: 05 Aug 2024

https://github.com/r9y9/deepvoice3_pytorch

PyTorch implementation of convolutional neural networks-based text-to-speech synthesis models

end-to-end machine-learning multi-speaker python pytorch speech-processing speech-synthesis tts

Last synced: 01 Aug 2024

https://github.com/deepcam-cn/yolov5-face

YOLO5Face: Why Reinventing a Face Detector (https://arxiv.org/abs/2105.12931) ECCV Workshops 2022)

arcface blazeface pytorch retinaface scrfd shufflenet tensorrt tinaface yolo yolov5 yolov7

Last synced: 31 Jul 2024

https://github.com/tristandeleu/pytorch-meta

A collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch

few-shot-learning meta-learning pytorch

Last synced: 01 Aug 2024

https://github.com/Hzzone/pytorch-openpose

pytorch implementation of openpose including Hand and Body Pose Estimation.

openpose pose-estimation pytorch

Last synced: 01 Aug 2024

https://github.com/jalammar/ecco

Explain, analyze, and visualize NLP language models. Ecco creates interactive visualizations directly in Jupyter notebooks explaining the behavior of Transformer-based language models (like GPT2, BERT, RoBERTA, T5, and T0).

explorables language-models natural-language-processing nlp pytorch visualization

Last synced: 01 Aug 2024

https://github.com/frgfm/torch-cam

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM)

activation-maps class-activation-map cnn deep-learning grad-cam gradcam gradcam-plus-plus interpretability interpretable-deep-learning python pytorch saliency-map score-cam smoothgrad

Last synced: 02 Aug 2024

https://github.com/albertpumarola/GANimation

GANimation: Anatomically-aware Facial Animation from a Single Image (ECCV'18 Oral) [PyTorch]

deep-learning eccv-2018 face-manipulation facial-expressions gan ganimation generative-adversarial-network pytorch

Last synced: 03 Aug 2024

https://github.com/DLR-RM/rl-baselines3-zoo

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

deep-reinforcement-learning gym hyperparameter-optimization hyperparameter-search hyperparameter-tuning lab openai optimization pybullet pybullet-environments pytorch reinforcement-learning rl robotics sde stable-baselines tuning-hyperparameters

Last synced: 02 Aug 2024

https://github.com/rguthrie3/DeepLearningForNLPInPytorch

An IPython Notebook tutorial on deep learning for natural language processing, including structure prediction.

deep-learning lstm neural-network nlp pytorch tutorial

Last synced: 01 Aug 2024

https://github.com/IDEA-Research/detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

anchor-detr conditional-detr dab-detr deformable-detr deta detr dino dn-detr group-detr h-detr mask-dino object-detection pose-estimation pytorch segmentation state-of-the-art

Last synced: 31 Jul 2024

https://github.com/learning-at-home/hivemind

Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.

asynchronous-programming asyncio deep-learning dht distributed-systems distributed-training hivemind machine-learning mixture-of-experts neural-networks pytorch volunteer-computing

Last synced: 01 Aug 2024

https://github.com/jiesutd/NCRFpp

NCRF++, a Neural Sequence Labeling Toolkit. Easy use to any sequence labeling tasks (e.g. NER, POS, Segmentation). It includes character LSTM/CNN, word LSTM/CNN and softmax/CRF components.

artificial-intelligence char-cnn char-rnn chunking cnn crf lstm lstm-crf named-entity-recognition natural-language-processing nbest ner neural-networks part-of-speech-tagger pytorch sequence-labeling

Last synced: 01 Aug 2024

https://github.com/labmlai/labml

🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱

analytics deep-learning experiment fastai keras keras-tensorflow machine-learning mobile pytorch pytorch-lightning tensorboard tensorflow tensorflow2 visualization

Last synced: 31 Jul 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: 03 Aug 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/open-mmlab/mmgeneration

MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.

diffusion-models gan generative generative-adversarial-network mmcv openmmlab pytorch

Last synced: 02 Aug 2024

https://github.com/nobody132/masr

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

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

Last synced: 02 Aug 2024

https://github.com/rlworkgroup/garage

A toolkit for reproducible reinforcement learning research.

pytorch reproducibility rl-algorithms tensorflow

Last synced: 31 Jul 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/jik876/hifi-gan

HiFi-GAN: Generative Adversarial Networks for Efficient and High Fidelity Speech Synthesis

deep-learning gan hifi-gan pytorch speech-synthesis text-to-speech tts vocoder

Last synced: 01 Aug 2024

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

OpenMMLab Rotated Object Detection Toolbox and Benchmark

detection openmmlab pytorch rotated-object

Last synced: 08 Aug 2024

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

PyTorch implementation of adversarial attacks [torchattacks].

adversarial-attacks deep-learning pytorch

Last synced: 31 Jul 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 工作流程和实验跟踪 80 Tutorials & books & examples｜教程 & 书籍 & 示例 80