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/swall0w/torchstat

Model analyzer in PyTorch

python pytorch

Last synced: 26 Sep 2024

https://github.com/V2AI/Det3D

World's first general purpose 3D object detection codebse.

3d-object-detection autonomous-driving deep-learning kitti nuscenes object-detection point-cloud pytorch

Last synced: 31 Jul 2024

https://github.com/poodarchu/Det3D

World's first general purpose 3D object detection codebse.

3d-object-detection autonomous-driving deep-learning kitti nuscenes object-detection point-cloud pytorch

Last synced: 13 Aug 2024

https://github.com/v2ai/det3d

World's first general purpose 3D object detection codebse.

3d-object-detection autonomous-driving deep-learning kitti nuscenes object-detection point-cloud pytorch

Last synced: 25 Sep 2024

https://github.com/erikwijmans/Pointnet2_PyTorch

PyTorch implementation of Pointnet2/Pointnet++

hydra point-cloud pointnet2 pytorch pytorch-lightning

Last synced: 31 Jul 2024

https://github.com/Tencent/TurboTransformers

a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.

albert bert decoder gpt2 gpu huggingface-transformers inference machine-translation nlp pytorch roberta transformer

Last synced: 31 Jul 2024

https://github.com/nvidiagameworks/kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).

3d-deep-learning nerf neuralfields pytorch sdf

Last synced: 25 Sep 2024

https://github.com/Swall0w/torchstat

Model analyzer in PyTorch

python pytorch

Last synced: 01 Aug 2024

https://github.com/marcoslucianops/deepstream-yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models

darknet deepstream nvidia nvidia-deepstream-sdk object-detection paddle paddlepaddle ppyoloe pytorch tensorrt ultralytics yolo yolor yolov3 yolov4 yolov5 yolov6 yolov7 yolov8 yolox

Last synced: 25 Sep 2024

https://github.com/toandaominh1997/efficientdet.pytorch

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

coco computer-vision demo detection efficientdet-d0 efficientnet focalloss multibox nms object-detection pascal-voc pytorch

Last synced: 25 Sep 2024

https://github.com/szagoruyko/attention-transfer

Improving Convolutional Networks via Attention Transfer (ICLR 2017)

attention deep-learning knowledge-distillation pytorch

Last synced: 25 Sep 2024

https://github.com/NVIDIAGameWorks/kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).

3d-deep-learning nerf neuralfields pytorch sdf

Last synced: 03 Aug 2024

https://github.com/toandaominh1997/EfficientDet.Pytorch

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

coco computer-vision demo detection efficientdet-d0 efficientnet focalloss multibox nms object-detection pascal-voc pytorch

Last synced: 07 Aug 2024

https://github.com/deepgraphlearning/torchdrug

A powerful and flexible machine learning platform for drug discovery

deep-learning drug-discovery graph-neural-networks pytorch

Last synced: 25 Sep 2024

https://github.com/filipradenovic/cnnimageretrieval-pytorch

CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch

cnn convolutional-neural-networks image-retrieval python pytorch

Last synced: 24 Sep 2024

https://github.com/keyu-tian/spark

[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"

bert cnn convnet convolutional-neural-networks deep-learning iclr iclr2023 instance-segmentation mae mask-rcnn masked-autoencoder masked-image-modeling object-detection pre-trained-model pretrain pretraining pytorch self-supervised-learning sparse-convolution ssl

Last synced: 24 Sep 2024

https://github.com/speedinghzl/ccnet

CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).

cityscape pytorch scene-parsing segmentation self-attention semantic-segmentation

Last synced: 24 Sep 2024

https://github.com/DeepGraphLearning/torchdrug

A powerful and flexible machine learning platform for drug discovery

deep-learning drug-discovery graph-neural-networks pytorch

Last synced: 02 Aug 2024

https://github.com/vturrisi/solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning

barlow-twins byol contrastive-learning deepcluster dino mae masked-input-prediction moco nnclr nvidia-dali pytorch pytorch-lightning ressl self-supervised-learning simclr simsiam swav transformer-models vibcreg vicreg

Last synced: 25 Sep 2024

https://github.com/nod-ai/shark-studio

SHARK Studio -- Web UI for SHARK+IREE High Performance Machine Learning Distribution

amd apple-silicon deep-learning machine-learning mlir nvidia pytorch

Last synced: 25 Sep 2024

https://github.com/speedinghzl/CCNet

CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).

cityscape pytorch scene-parsing segmentation self-attention semantic-segmentation

Last synced: 31 Jul 2024

https://github.com/goatmessi8/rfbnet

Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018

detection mobilenet pytorch rfbnet

Last synced: 24 Sep 2024

https://github.com/keyu-tian/SparK

[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"

bert cnn convnet convolutional-neural-networks deep-learning iclr iclr2023 instance-segmentation mae mask-rcnn masked-autoencoder masked-image-modeling object-detection pre-trained-model pretrain pretraining pytorch self-supervised-learning sparse-convolution ssl

Last synced: 31 Jul 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: 24 Sep 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: 02 Aug 2024

https://github.com/cszn/DnCNN

Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)

image-denoising jpeg-deblocking keras-tensorflow matconvnet pytorch residual-learning super-resolution

Last synced: 01 Aug 2024

https://github.com/GOATmessi7/RFBNet

Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018

detection mobilenet pytorch rfbnet

Last synced: 01 Aug 2024

https://github.com/torchgan/torchgan

Research Framework for easy and efficient training of GANs based on Pytorch

computer-vision deep-learning gans generative-adversarial-networks generative-model machine-learning neural-networks python python3 pytorch

Last synced: 01 Aug 2024

https://github.com/JiaRenChang/PSMNet

Pyramid Stereo Matching Network (CVPR2018)

psmnet pytorch stereo-matching stereo-vision

Last synced: 31 Jul 2024

https://github.com/qfgaohao/pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

object-detection open-images pytorch ssd

Last synced: 24 Sep 2024

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

Pre-trained Chinese ELECTRA(中文ELECTRA预训练模型)

bert chinese chinese-electra electra language-model nlp pre-trained-model pytorch tensorflow

Last synced: 01 Aug 2024

https://github.com/patrick-kidger/torchtyping

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.

named-tensors python-typing pytorch shape tensors typing

Last synced: 24 Sep 2024

https://github.com/nod-ai/SHARK

SHARK - High Performance Machine Learning Distribution

amd apple-silicon deep-learning machine-learning mlir nvidia pytorch

Last synced: 01 Aug 2024

https://github.com/nerfstudio-project/nerfacc

A General NeRF Acceleration Toolbox in PyTorch.

instant-ngp nerf pytorch rendering

Last synced: 24 Sep 2024

https://github.com/pytorch/functorch

functorch is JAX-like composable function transforms for PyTorch.

gradients hessians pytorch

Last synced: 24 Sep 2024

https://github.com/bermanmaxim/lovaszsoftmax

Code for the Lovász-Softmax loss (CVPR 2018)

image-segmentation loss-functions neural-networks pytorch

Last synced: 24 Sep 2024

https://github.com/n00mkrad/flowframes

Flowframes Windows GUI for video interpolation using DAIN (NCNN) or RIFE (CUDA/NCNN)

dain gui ncnn pytorch rife video-interpolation

Last synced: 31 Jul 2024

https://github.com/dmmiller612/bert-extractive-summarizer

Easy to use extractive text summarization with BERT

bert coreference extractive-summarization pytorch

Last synced: 25 Sep 2024

https://github.com/BenAAndrew/Voice-Cloning-App

A Python/Pytorch app for easily synthesising human voices

deep-learning python pytorch tacotron2 text-to-speech tts voice-cloning

Last synced: 31 Jul 2024

https://github.com/UKPLab/beir

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.

ance benchmark bert colbert dataset deep-learning dpr elasticsearch information-retrieval nlp passage-retrieval pytorch question-generation retrieval retrieval-models sbert sentence-transformers use-qa zero-shot-retrieval

Last synced: 04 Aug 2024

https://github.com/beir-cellar/beir

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.

ance benchmark bert colbert dataset deep-learning dpr elasticsearch information-retrieval nlp passage-retrieval pytorch question-generation retrieval retrieval-models sbert sentence-transformers use-qa zero-shot-retrieval

Last synced: 25 Sep 2024

https://github.com/diffeqml/torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

control-theory deep-equilibrium-models deep-learning dynamical-systems implicit-models neural-differential-equations neural-network neural-ode numerical-methods pytorch

Last synced: 25 Sep 2024

https://github.com/deepmodeling/deepmd-kit

A deep learning package for many-body potential energy representation and molecular dynamics

ase c computational-chemistry cpp cuda deep-learning deepmd ipi lammps materials-science molecular-dynamics nodejs potential-energy python pytorch rocm tensorflow

Last synced: 25 Sep 2024

https://github.com/yoshitomo-matsubara/torchdistill

A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆25 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.

amazon-sagemaker-lab cifar10 cifar100 coco colab-notebook glue google-colab image-classification imagenet knowledge-distillation natural-language-processing nlp object-detection pascal-voc pytorch semantic-segmentation transformer

Last synced: 25 Sep 2024

https://github.com/lxztju/pytorch_classification

利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码

cnn densenet deployment flask image-classification knn knowledge-distillation label-smoothing pytorch random-forest resnet resnext svm

Last synced: 24 Sep 2024

https://github.com/trademaster-ntu/trademaster

TradeMaster is an open-source platform for quantitative trading empowered by reinforcement learning :fire: :zap: :rainbow:

finance fintech investment-strategies jupyter-notebook machine-learning python pytorch quantitative-trading reinforcement-learning stock-market trading-platform

Last synced: 26 Sep 2024

https://github.com/arthur151/romp

Monocular, One-stage, Regression of Multiple 3D People and their 3D positions & trajectories in camera & global coordinates. ROMP[ICCV21], BEV[CVPR22], TRACE[CVPR2023]

3d-mesh-recovery bottom-up multi-person multi-person-3d-mesh-recovery pose-estimation pytorch smpl

Last synced: 25 Sep 2024

https://github.com/cysu/open-reid

Open source person re-identification library in python

deep-learning metric-learning person-reidentification python pytorch

Last synced: 24 Sep 2024

https://github.com/vitae-transformer/vitpose

The official repo for [NeurIPS'22] "ViTPose: Simple Vision Transformer Baselines for Human Pose Estimation" and [TPAMI'23] "ViTPose++: Vision Transformer for Generic Body Pose Estimation"

deep-learning distillation mae pose-estimation pytorch self-supervised-learning vision-transformer

Last synced: 24 Sep 2024

https://github.com/intel/intel-extension-for-pytorch

A Python package for extending the official PyTorch that can easily obtain performance on Intel platform

deep-learning intel machine-learning neural-network pytorch quantization

Last synced: 25 Sep 2024

https://github.com/Arthur151/ROMP

Monocular, One-stage, Regression of Multiple 3D People and their 3D positions & trajectories in camera & global coordinates. ROMP[ICCV21], BEV[CVPR22], TRACE[CVPR2023]

3d-mesh-recovery bottom-up multi-person multi-person-3d-mesh-recovery pose-estimation pytorch smpl

Last synced: 31 Jul 2024

https://github.com/d-li14/involution

[CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator

cvpr2021 image-classification instance-segmentation involution object-detection operator pre-trained-model pytorch semantic-segmentation

Last synced: 24 Sep 2024

https://github.com/miracleyoo/pytorch-lightning-template

An easy/swift-to-adapt PyTorch-Lighting template. 套壳模板,简单易用,稍改原来Pytorch代码,即可适配Lightning。You can translate your previous Pytorch code much easier using this template, and keep your freedom to edit all the functions as well. Big-project-friendly as well. No need to rewrite your config in hydra.

best-practices config deep-learning pytorch pytorch-lightning pytorch-lightning-template template

Last synced: 26 Sep 2024

https://github.com/fangwei123456/spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

deep-learning dvs machine-learning pytorch snn spiking-neural-networks

Last synced: 25 Sep 2024

https://github.com/nianticlabs/simplerecon

[ECCV 2022] SimpleRecon: 3D Reconstruction Without 3D Convolutions

computer-vision cost-volume depth depth-estimation eccv2022 multi-view-stereo mvs pytorch scannet visualization

Last synced: 24 Sep 2024

https://github.com/stared/livelossplot

Live training loss plot in Jupyter Notebook for Keras, PyTorch and others

deep-learning jupyter-notebook keras keras-visualization pytorch

Last synced: 25 Sep 2024

https://github.com/takuseno/d3rlpy

An offline deep reinforcement learning library

deep-learning deep-reinforcement-learning offline-rl pytorch

Last synced: 25 Sep 2024

https://github.com/DiffEqML/torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

control-theory deep-equilibrium-models deep-learning dynamical-systems implicit-models neural-differential-equations neural-network neural-ode numerical-methods pytorch

Last synced: 31 Jul 2024

https://github.com/sail-sg/poolformer

PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)

image-classification mlp pooling pytorch transformer

Last synced: 24 Sep 2024

https://github.com/charlesshang/dcnv2

Deformable Convolutional Networks v2 with Pytorch

cnn deformable-convolutional-networks pytorch

Last synced: 24 Sep 2024

https://github.com/920232796/bert_seq2seq

pytorch实现 Bert 做seq2seq任务,使用unilm方案,现在也可以做自动摘要,文本分类,情感分析,NER,词性标注等任务,支持t5模型,支持GPT2进行文章续写。

autotitle bert crf gpt2 ner pytorch roberta seq2seq t5-model text-classification unilm

Last synced: 24 Sep 2024

https://github.com/jayleicn/animeGAN

A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.

dataset generative-adversarial-network pytorch

Last synced: 02 Aug 2024

https://github.com/jayleicn/animegan

A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.

dataset generative-adversarial-network pytorch

Last synced: 24 Sep 2024

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

:couple: Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) :couple:

apex cuhk-np dg-net dukemtmc-reid image-retrieval image-search market-1501 msmt17 open-reid person-reid person-reidentification pytorch re-identification

Last synced: 25 Sep 2024

https://github.com/NVlabs/DG-Net

:couple: Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) :couple:

apex cuhk-np dg-net dukemtmc-reid image-retrieval image-search market-1501 msmt17 open-reid person-reid person-reidentification pytorch re-identification

Last synced: 01 Aug 2024

https://github.com/godweiyang/NN-CUDA-Example

Several simple examples for popular neural network toolkits calling custom CUDA operators.

cpp cuda neural-network python pytorch tensorflow

Last synced: 31 Jul 2024

https://github.com/alibaba/federatedscope

An easy-to-use federated learning platform

federated-learning machine-learning pytorch

Last synced: 25 Sep 2024

https://github.com/jeshraghian/snntorch

Deep and online learning with spiking neural networks in Python

machine-learning neural-networks neuron-models neuroscience python pytorch snn spike spiking spiking-neural-networks

Last synced: 25 Sep 2024

https://github.com/hiyouga/fastedit

🩹Editing large language models within 10 seconds⚡

bloom chatbots chatgpt falcon gpt large-language-models llama llms pytorch transformers

Last synced: 24 Sep 2024

https://github.com/WisconsinAIVision/yolact_edge

The first competitive instance segmentation approach that runs on small edge devices at real-time speeds.

edge-devices instance-segmentation pytorch real-time realtime yolactedge

Last synced: 31 Jul 2024