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/yinchangchang/ocr_densenet

第一届西安交通大学人工智能实践大赛(2018AI实践大赛--图片文字识别)第一名;仅采用densenet识别图中文字

densenet ocr ocr-recognition python pytorch

Last synced: 03 Nov 2024

https://github.com/Schlumberger/joint-vae

Pytorch implementation of JointVAE, a framework for disentangling continuous and discrete factors of variation :star2:

disentangled-representations disentanglement generative-models gumbel-softmax pytorch vae

Last synced: 03 Aug 2024

https://github.com/EdisonLeeeee/GraphGallery

GraphGallery is a gallery for benchmarking Graph Neural Networks, From InplusLab.

adversarial-machine-learning dgl geometric-deep-learning graph graph-neural-networks graphgallery models pyg pytorch

Last synced: 02 Aug 2024

https://github.com/layumi/AICIty-reID-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)

aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776

Last synced: 08 Nov 2024

https://github.com/zalandoresearch/pytorch-vq-vae

PyTorch implementation of VQ-VAE by Aäron van den Oord et al.

deep-learning pytorch vae vq-vae

Last synced: 10 Nov 2024

https://github.com/layumi/aicity-reid-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)

aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776

Last synced: 31 Oct 2024

https://github.com/walktree/libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm

cpp libtorch pytorch yolov3

Last synced: 09 Nov 2024

https://github.com/clovaai/rexnet

Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models

deep-learning image-classification imagenet object-detection pretrained-model pytorch rexnet transfer-learning

Last synced: 07 Aug 2024

https://github.com/sinAshish/Multi-Scale-Attention

Code for our paper "Multi-scale Guided Attention for Medical Image Segmentation"

attention-mechanism biomedical-image-analysis deep-learning encoder-decoder pytorch

Last synced: 03 Aug 2024

https://github.com/huggingface/large_language_model_training_playbook

An open collection of implementation tips, tricks and resources for training large language models

cuda large-language-models llm nccl nlp performance python pytorch scalability troubleshooting

Last synced: 11 Nov 2024

https://github.com/dfdazac/wassdistance

Approximating Wasserstein distances with PyTorch

deep-learning machine-learning optimal-transport pytorch

Last synced: 02 Aug 2024

https://github.com/ploomber/sklearn-evaluation

Machine learning model evaluation made easy: plots, tables, HTML reports, experiment tracking and Jupyter notebook analysis.

data-science deep-learning jupyter-notebook machine-learning pytorch scikit-learn sklearn tensorflow

Last synced: 13 Oct 2024

https://github.com/cuiziteng/Illumination-Adaptive-Transformer

[BMVC 2022] You Only Need 90K Parameters to Adapt Light: A Light Weight Transformer for Image Enhancement and Exposure Correction. SOTA for low light enhancement, 0.004 seconds try this for pre-processing.

bmvc exposure-correction image-enhancement image-reconstruction image-restoration low-level-vision low-light-enhance low-light-image-enhancement pytorch transformer-architecture transformer-models

Last synced: 06 Nov 2024

https://github.com/wvangansbeke/sparse-depth-completion

Predict dense depth maps from sparse and noisy LiDAR frames guided by RGB images. (Ranked 1st place on KITTI) [2019]

computer-vision deep-learning depth-completion depth-prediction kitti lidar noisy-data pytorch sensor-fusion

Last synced: 07 Nov 2024

https://github.com/chaitjo/geometric-gnn-dojo

Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks.

e3nn equivariance equivariant-networks geometric-deep-learning geometric-graphs graph-isomorphism graph-neural-networks pytorch pytorch-geometric

Last synced: 01 Nov 2024

https://github.com/loretoparisi/CapsNet

CapsNet (Capsules Net) in Geoffrey E Hinton paper "Dynamic Routing Between Capsules" - State Of the Art

capsnet capsule-networks chainer deeplearning keras lasagne machine-learning mnist mxnet pytorch tensorflow

Last synced: 02 Nov 2024

https://github.com/RaivoKoot/Video-Dataset-Loading-Pytorch

Generic PyTorch dataset implementation to load and augment VIDEOS for deep learning training loops.

action-recognition dataloader deep-learning machine-learning pytorch video-dataset videos

Last synced: 10 Nov 2024

https://github.com/mv-lab/InstructIR

[ECCV 2024] InstructIR: High-Quality Image Restoration Following Human Instructions https://huggingface.co/spaces/marcosv/InstructIR

computer-vision deblurring deep-learning dehazing denoising image-enhancement image-restoration inverse-problems language-model low-light-image-enhancement multi-task multimodal neural-network photography prompt pytorch super-resolution

Last synced: 03 Nov 2024

https://github.com/sacmehta/ESPNetv2

A light-weight, power efficient, and general purpose convolutional neural network

cnn convolutional-neural-networks deep-learning efficient imagenet lightweight machine-learning pytorch semantic-segmentation

Last synced: 31 Oct 2024

https://github.com/hkchengrex/mivos

[CVPR 2021] Modular Interactive Video Object Segmentation: Interaction-to-Mask, Propagation and Difference-Aware Fusion. Semi-supervised VOS as well!

computer-vision cvpr2021 deep-learning interactive-segmentation pytorch segmentation video-object-segmentation video-segmentation

Last synced: 10 Nov 2024

https://github.com/iBelieveCJM/Tricks-of-Semi-supervisedDeepLeanring-Pytorch

PseudoLabel 2013, VAT, PI model, Tempens, MeanTeacher, ICT, MixMatch, FixMatch

fixmatch ict meanteacher mixmatch pi pseudo-label pytorch semi-supervised-learning ssl tempens vat

Last synced: 03 Aug 2024

https://github.com/sthalles/pytorch-byol

PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning

byol deep-learning pytorch representation-learning self-supervised-learning simclr unsupervised-learning

Last synced: 06 Nov 2024

https://github.com/THUDM/GraphMAE

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

deep-learning graph-neural-networks pytorch self-supervised-learning

Last synced: 08 Aug 2024

https://github.com/ruiminshen/yolo2-pytorch

PyTorch implementation of the YOLO (You Only Look Once) v2

caffe2 computer-vision deep-learning deep-neural-networks object-detection onnx python pytorch yolo2

Last synced: 04 Nov 2024

https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2

Ultra Fast Deep Lane Detection With Hybrid Anchor Driven Ordinal Classification (TPAMI 2022)

deep-learning lane-detection pytorch

Last synced: 11 Nov 2024

https://github.com/lemonhu/ner-bert-pytorch

PyTorch solution of named entity recognition task Using Google AI's pre-trained BERT model.

chinese-ner entity-extraction google-bert information-extraction msra named-entity-recognition ner pytorch transformer

Last synced: 08 Nov 2024

https://github.com/nschaetti/EchoTorch

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.

artificial-intelligence artificial-neural-networks echo-state-networks machine-learning machine-learning-algorithms machinelearning neural-networks python python-toolkit pytorch recurrent-networks recurrent-neural-networks reservoir-computing torch

Last synced: 03 Aug 2024

https://github.com/pykale/pykale

Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for interdisciplinary research, part of the 🔥PyTorch ecosystem. ⭐ Star to support our work!

computer-vision data-science deep-learning domain-adaptation graph-analysis knowledge-aware-learning machine-learning medical-image-analysis meta-learning multimodal multimodal-learning python pytorch transfer-learning

Last synced: 11 Oct 2024

https://github.com/lemonhu/NER-BERT-pytorch

PyTorch solution of named entity recognition task Using Google AI's pre-trained BERT model.

chinese-ner entity-extraction google-bert information-extraction msra named-entity-recognition ner pytorch transformer

Last synced: 02 Nov 2024

https://github.com/aamini/evidential-deep-learning

Learn fast, scalable, and calibrated measures of uncertainty using neural networks!

confidence deep-learning evidence neural-network neurips-2020 pytorch tensorflow uncertainty

Last synced: 30 Oct 2024

https://github.com/tlkh/ai-lab

All-in-one AI container for rapid prototyping

cuda data-science deep-learning docker jupyter nvidia pytorch tensorflow

Last synced: 07 Nov 2024

https://github.com/daniilrobnikov/vits2

VITS2: Improving Quality and Efficiency of Single-Stage Text-to-Speech with Adversarial Learning and Architecture Design

deep-learning pytorch speech speech-synthesis text-to-speech tts vits2 voice-conversion

Last synced: 27 Oct 2024

https://github.com/likyoo/open-cd

A Change Detection Repo Standing on the Shoulders of Giants

bit change-detection changer deep-learning pytorch transformer

Last synced: 03 Aug 2024

https://github.com/skhadem/3D-BoundingBox

PyTorch implementation for 3D Bounding Box Estimation Using Deep Learning and Geometry

3d computer-vision localization pytorch

Last synced: 28 Oct 2024

https://github.com/cccntu/minlora

minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.

fastai huggingface pytorch pytorch-implementation pytorch-lightning

Last synced: 09 Nov 2024

https://github.com/M-Nauta/TCDF

Temporal Causal Discovery Framework (PyTorch): discovering causal relationships between time series

causality cnn forecasting machine-learning neural-networks prediction python pytorch timeseries

Last synced: 03 Aug 2024

https://github.com/kubeflow/mpi-operator

Kubernetes Operator for MPI-based applications (distributed training, HPC, etc.)

apache-mxnet distributed-computing horovod kubeflow kubernetes mpi pytorch tensorflow

Last synced: 14 Oct 2024

https://github.com/ashishpatel26/Real-time-ML-Project

A curated list of applied machine learning and data science notebooks and libraries across different industries.

application deep-learning deeplearning dl keras machine-learning machine-learning-algorithms machinelearning ml ml-application project pytorch real-time real-time-data rl tensorflow theano

Last synced: 01 Nov 2024

https://github.com/um-arm-lab/pytorch_mppi

Model Predictive Path Integral (MPPI) with approximate dynamics implemented in pytorch

approximate-dynamics controls model-predictive-control mppi pytorch

Last synced: 09 Nov 2024

https://github.com/shm007g/LLaMA-Cult-and-More

Large Language Models for All, 🦙 Cult and More, Stay in touch !

alpaca chatgpt deepspeed ggml gpt gpt4 gptq llama llm loralib pytorch tensorflow transformers vicuna

Last synced: 27 Oct 2024

https://github.com/KinglittleQ/torch-batch-svd

A 100x faster SVD for PyTorch⚡️

cusolver pytorch svd

Last synced: 27 Oct 2024

https://github.com/allentdan/libtorchsegmentation

A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.

cpp deeplabv3 deeplabv3plus fpn image-segmentation imagenet libtorch libtorch-segment models neural-network pretrained-backbones pretrained-weights pspnet pytorch pytorch-cpp pytorch-cpp-frontend resnet resnext semantic-segmentation unet

Last synced: 26 Oct 2024

https://github.com/open-compass/VLMEvalKit

Open-source evaluation toolkit of large vision-language models (LVLMs), support GPT-4v, Gemini, QwenVLPlus, 40+ HF models, 20+ benchmarks

chatgpt claude clip computer-vision evaluation gemini gpt gpt-4v gpt4 large-language-models llava llm multi-modal openai openai-api pytorch qwen vit vqa

Last synced: 08 Aug 2024

https://github.com/zihangJiang/TokenLabeling

Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"

imagenet lv-vit pytorch segmentation transformer vision

Last synced: 02 Aug 2024

https://github.com/adefossez/julius

Fast PyTorch based DSP for audio and 1D signals

dsp pytorch

Last synced: 20 Oct 2024

https://github.com/Eromera/erfnet_pytorch

Pytorch code for semantic segmentation using ERFNet

cityscapes erfnet pytorch segmentation semantic semantic-segmentation

Last synced: 03 Aug 2024

https://github.com/songquanpeng/pytorch-template

To be the world's best PyTorch project template.

boilerplates gan gans pytorch pytorch-template

Last synced: 30 Oct 2024

https://github.com/shinya7y/UniverseNet

USB: Universal-Scale Object Detection Benchmark (BMVC 2022)

mmdetection object-detection pytorch waymo-open-dataset

Last synced: 28 Oct 2024

https://github.com/abhiskk/fast-neural-style

pytorch implementation of fast-neural-style

deep-learning neural-style pytorch

Last synced: 06 Nov 2024

https://github.com/Stanford-TML/EDGE

Official PyTorch Implementation of EDGE (CVPR 2023)

animation dance-generation diffusion-models pytorch

Last synced: 03 Aug 2024

https://github.com/eliahuhorwitz/DeepSIM

Official PyTorch implementation of the paper: "DeepSIM: Image Shape Manipulation from a Single Augmented Training Sample" (ICCV 2021 Oral)

computer-graphics computer-vision deep-learning deep-neural-networks edge-to-image generative-adversarial-network iccv2021 image-editing image-to-image-translation pytorch segmantation-to-image single-image

Last synced: 07 Aug 2024

https://github.com/Jeff-sjtu/res-loglikelihood-regression

Code for "Human Pose Regression with Residual Log-likelihood Estimation", ICCV 2021 Oral

2d-human-pose 3d-human-pose human-pose-estimation iccv iccv2021 pytorch regression

Last synced: 03 Aug 2024

https://github.com/eliahuhorwitz/deepsim

Official PyTorch implementation of the paper: "DeepSIM: Image Shape Manipulation from a Single Augmented Training Sample" (ICCV 2021 Oral)

computer-graphics computer-vision deep-learning deep-neural-networks edge-to-image generative-adversarial-network iccv2021 image-editing image-to-image-translation pytorch segmantation-to-image single-image

Last synced: 10 Nov 2024

https://github.com/NasirKhalid24/CLIP-Mesh

Official implementation of CLIP-Mesh: Generating textured meshes from text using pretrained image-text models

3d pytorch

Last synced: 06 Nov 2024

https://github.com/blackhc/toma

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory

data-science gpu machine-learning python pytorch

Last synced: 09 Nov 2024

https://github.com/interpretml/interpret-community

Interpret Community extends Interpret repository with additional interpretability techniques and utility functions to handle real-world datasets and workflows.

deep-learning explainer explanationdashboard interpretable-models machine-learning mimic-explainer python pytorch tabular-explainer

Last synced: 10 Nov 2024

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

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations neural-networks pytorch time-series

Last synced: 29 Oct 2024

https://github.com/naver/roma

RoMa: A lightweight library to deal with 3D rotations in PyTorch.

3d geometry pytorch rotation

Last synced: 08 Nov 2024

https://naver.github.io/roma/

RoMa: A lightweight library to deal with 3D rotations in PyTorch.

3d geometry pytorch rotation

Last synced: 03 Aug 2024

https://github.com/petuum/adaptdl

Resource-adaptive cluster scheduler for deep learning training.

aws cloud deep-learning distributed-systems distributed-training kubernetes machine-learning pytorch

Last synced: 03 Aug 2024

https://github.com/krasserm/perceiver-io

A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training

deep-learning machine-learning perceiver perceiver-ar perceiver-io pytorch pytorch-lightning

Last synced: 29 Oct 2024

https://github.com/continvvm/continuum

A clean and simple data loading library for Continual Learning

continual-learning dataloader dataset incremental-learning lifelong-learning online-learning pytorch

Last synced: 29 Oct 2024

https://github.com/hysts/anime-face-detector

Anime Face Detector using mmdet and mmpose

anime computer-vision face-detection face-landmark-detection pytorch

Last synced: 30 Oct 2024

https://github.com/Lotayou/densebody_pytorch

PyTorch implementation of CloudWalk's recent work DenseBody

3d-reconstruction human pose-estimation pytorch

Last synced: 04 Nov 2024

https://github.com/ethanhe42/epipolar-transformers

Epipolar Transformers (best paper award, CVPR 2020 workshop)

3d 3dposeestimation deep-learning pose-estimation pytorch

Last synced: 30 Oct 2024

https://github.com/cccntu/minLoRA

minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.

fastai huggingface pytorch pytorch-implementation pytorch-lightning

Last synced: 24 Oct 2024

https://github.com/clovaai/AdamP

AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)

deep-learning iclr2021 machine-learning optimizer optimizer-algorithms pytorch

Last synced: 03 Aug 2024

https://github.com/greydanus/hamiltonian-nn

Code for our paper "Hamiltonian Neural Networks"

deep-learning dynamics hamiltonian neural-network physics pytorch research

Last synced: 30 Oct 2024

https://github.com/apachecn/fastai-ml-dl-notes-zh

:book: [译] fast.ai 机器学习和深度学习中文笔记

deep-learning fastai lecture-notes machine-learning python pytorch

Last synced: 29 Oct 2024

https://github.com/Media-Smart/vedaseg

A semantic segmentation toolbox based on PyTorch

deeplabv3 deeplabv3plus fpn pspnet pytorch semantic-segmentation unet

Last synced: 03 Aug 2024

https://github.com/MuQiuJun-AI/bert4pytorch

超轻量级bert的pytorch版本,大量中文注释,容易修改结构,持续更新

bert nlp pytorch transformer

Last synced: 06 Nov 2024

https://github.com/AdamCobb/hamiltorch

PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks

bnn hmc pytorch

Last synced: 30 Oct 2024

https://github.com/nvidia-merlin/dataloader

The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX

deep-learning jax pytorch recommender-systems tensorflow

Last synced: 08 Nov 2024