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/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: 29 Oct 2024

https://github.com/sthalles/simclr

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

contrastive-loss deep-learning machine-learning pytorch pytorch-implementation representation-learning simclr torchvision unsupervised-learning

Last synced: 15 Oct 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/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: 09 Nov 2024

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

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

deep-learning llama2 llama2-docker llm pytorch

Last synced: 25 Oct 2024

https://github.com/linksoul-ai/chinese-llama-2-7b

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

deep-learning llama2 llama2-docker llm pytorch

Last synced: 10 Oct 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: 15 Oct 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: 09 Oct 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: 27 Oct 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: 09 Oct 2024

https://github.com/taesungp/contrastive-unpaired-translation

Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)

computer-graphics computer-vision computervision cyclegan deeplearning gans generative-adversarial-network image-generation image-manipulation pytorch

Last synced: 15 Oct 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: 29 Oct 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: 28 Oct 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: 15 Nov 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: 15 Oct 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 Nov 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: 15 Nov 2024

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

Pytorch implementation of RetinaNet object detection.

python pytorch retinanet

Last synced: 25 Sep 2024

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

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

deep-learning python python3 pytorch tensorflow

Last synced: 30 Oct 2024

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

Machine learning metrics for distributed, scalable PyTorch applications.

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

Last synced: 30 Oct 2024

https://github.com/PyTorchLightning/metrics

Machine learning metrics for distributed, scalable PyTorch applications.

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

Last synced: 08 Nov 2024

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

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

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

Last synced: 03 Nov 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: 06 Nov 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: 14 Oct 2024

https://github.com/daniil-osokin/lightweight-human-pose-estimation.pytorch

Fast and accurate human pose estimation in PyTorch. Contains implementation of "Real-time 2D Multi-Person Pose Estimation on CPU: Lightweight OpenPose" paper.

coco-keypoints-detection deep-learning human-pose-estimation lightweight lightweight-openpose mscoco-keypoint openpose openvino pose-estimation pytorch real-time

Last synced: 14 Nov 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: 15 Oct 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: 16 Nov 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: 14 Oct 2024

https://github.com/jack-cherish/deep-learning

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

deep-learning python python3 pytorch tensorflow

Last synced: 18 Oct 2024

https://github.com/aizootech/facemaskdetection

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

caffe detection pytorch

Last synced: 15 Oct 2024

https://github.com/AIZOOTech/FaceMaskDetection

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

caffe detection pytorch

Last synced: 27 Oct 2024

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

pytorch implementation of openpose including Hand and Body Pose Estimation.

openpose pose-estimation pytorch

Last synced: 14 Oct 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: 27 Oct 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: 14 Oct 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: 12 Nov 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: 14 Oct 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: 15 Oct 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: 28 Oct 2024

https://github.com/ashawkey/torch-ngp

A pytorch CUDA extension implementation of instant-ngp (sdf and nerf), with a GUI.

d-nerf gui instant-ngp nerf pytorch real-time sdf tensorf

Last synced: 12 Nov 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: 22 Oct 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: 15 Oct 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: 29 Oct 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: 14 Oct 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: 28 Oct 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: 09 Oct 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: 29 Oct 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: 15 Oct 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: 14 Oct 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: 26 Oct 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: 12 Oct 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: 15 Oct 2024

https://github.com/shibing624/python-tutorial

Python实用教程,包括:Python基础,Python高级特性,面向对象编程,多线程,数据库,数据科学,Flask,爬虫开发教程。

numpy pandas python python-tutorial pytorch scikit-learn scipy

Last synced: 22 Oct 2024

https://github.com/cure-lab/ltsf-linear

[AAAI-23 Oral] Official implementation of the paper "Are Transformers Effective for Time Series Forecasting?"

aaai aaai2023 deep-learning forecasting forecasting-model linear-models pytorch time-series time-series-forecasting time-series-prediction

Last synced: 14 Oct 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: 13 Oct 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: 15 Oct 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: 11 Oct 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: 25 Sep 2024

https://github.com/vainf/deeplabv3plus-pytorch

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes

cityscapes deeplabv3 deeplabv3plus pascal-voc pytorch segmentation

Last synced: 14 Oct 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: 15 Oct 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: 15 Nov 2024

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

pytorch implementation of openpose including Hand and Body Pose Estimation.

openpose pose-estimation pytorch

Last synced: 03 Nov 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: 02 Nov 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: 14 Oct 2024

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

An unofficial styleguide and best practices summary for PyTorch

best-practices pytorch styleguide

Last synced: 12 Nov 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: 15 Oct 2024

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

An unofficial styleguide and best practices summary for PyTorch

best-practices pytorch styleguide

Last synced: 14 Oct 2024

https://github.com/pytorch/torchrec

Pytorch domain library for recommendation systems

cuda deep-learning gpu pytorch recommendation-system recommender-system sharding

Last synced: 29 Oct 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: 06 Nov 2024

https://github.com/breezedeus/Pix2Text

An Open-Source Python3 tool for recognizing layouts, tables, math formulas (LaTeX), and text in images, converting them into Markdown format. A free alternative to Mathpix, empowering seamless conversion of visual content into text-based representations. 80+ languages are supported.

image-to-markdown latex latex-pdf layout-analysis math-formula math-formula-recognition math-ocr mathpix ocr python pytorch table-ocr

Last synced: 09 Nov 2024