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/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: 25 Dec 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: 26 Dec 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: 21 Dec 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: 21 Dec 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: 25 Sep 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: 23 Nov 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/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: 21 Dec 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: 20 Dec 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/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: 26 Dec 2024

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

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

deep-learning python python3 pytorch tensorflow

Last synced: 21 Dec 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/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: 26 Dec 2024

https://github.com/predibase/lorax

Multi-LoRA inference server that scales to 1000s of fine-tuned LLMs

fine-tuning gpt llama llm llm-inference llm-serving llmops lora model-serving pytorch transformers

Last synced: 24 Dec 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: 20 Dec 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/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: 24 Dec 2024

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

pytorch implementation of openpose including Hand and Body Pose Estimation.

openpose pose-estimation pytorch

Last synced: 26 Dec 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: 23 Dec 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/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: 19 Dec 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: 21 Dec 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/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: 27 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: 21 Dec 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: 23 Dec 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: 21 Dec 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: 20 Dec 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/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: 24 Dec 2024

https://github.com/aizootech/facemaskdetection

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

caffe detection pytorch

Last synced: 21 Dec 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/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: 24 Dec 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: 24 Dec 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/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/vainf/deeplabv3plus-pytorch

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes

cityscapes deeplabv3 deeplabv3plus pascal-voc pytorch segmentation

Last synced: 21 Dec 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: 24 Dec 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: 20 Dec 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: 26 Dec 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: 24 Dec 2024

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

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

numpy pandas python python-tutorial pytorch scikit-learn scipy

Last synced: 20 Dec 2024

https://github.com/nvidia/transformerengine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

cuda deep-learning fp8 gpu jax machine-learning python pytorch

Last synced: 26 Dec 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/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: 21 Dec 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: 25 Dec 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: 26 Dec 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: 21 Dec 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: 24 Dec 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/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: 26 Dec 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/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: 21 Dec 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/igorsusmelj/pytorch-styleguide

An unofficial styleguide and best practices summary for PyTorch

best-practices pytorch styleguide

Last synced: 21 Dec 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: 21 Dec 2024

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

PyTorch implementation of adversarial attacks [torchattacks]

adversarial-attacks deep-learning pytorch

Last synced: 26 Dec 2024