https://github.com/ensta-u2is-ai/torch-uncertainty
Open-source framework for uncertainty and deep learning models in PyTorch :seedling:
bayesian-network
computer-vision
ensembles
mixup
neural-networks
predictive-uncertainty
pytorch
reliable-ai
trustworthy-machine-learning
uncertainty
uncertainty-quantification
Added: over 1 year ago - Last Synced: 11 months ago
- Created: February 01, 2023
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 783
- Committers: 11
- Issues: 23
- Pull Requests: 77
- Owner: ENSTA-U2IS-AI
- Stars: 195
- Forks: 16
- Packages: 0

https://github.com/ze-yang/lgkd
[ICCV 2023] Official implementation for "Label-Guided Knowledge Distillation for Continual Semantic Segmentation on 2D Images and 3D Point Clouds".
3d-point-clouds
continual-learning
continual-semantic-segmentation
knowledge-distillation
pytorch
semantic-segmentation
Added: over 1 year ago - Last Synced: 11 months ago
- Created: August 18, 2023

https://github.com/noisereduce/torchspectralgating
TorchSpectralGate is a PyTorch-based implementation of Spectral Gating, an algorithm for denoising audio signals.
acoustics
audio
enhancement
noise-reduction
parallel-computing
pytorch
signal-processing
speech
time-domain
torch
Added: over 1 year ago - Last Synced: 11 months ago
- Created: April 01, 2023
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 55
- Committers: 1
- Issues: 1
- Pull Requests: 2
- Owner: noisereduce
- Stars: 16
- Forks: 0
- Packages: 1
- Downloads: 61

https://github.com/levan92/deep_sort_realtime
A really more real-time adaptation of deep sort
computer-vision
deep-sort-tracking
deepsort
multi-object-tracking
pytorch
tracking
Added: over 1 year ago - Last Synced: 11 months ago
- Created: July 05, 2019

https://github.com/aican-research/fast-pathology
β‘ Open-source software for deep learning-based digital pathology
computational-pathology
convolutional-neural-networks
cplusplus
deep-learning
deployment
digital-pathology
fastpathology
free-to-use
inference
machine-learning
opencl
opengl
openvino
pytorch
software
tensorflow
tensorrt
Added: over 1 year ago - Last Synced: 11 months ago
- Created: July 09, 2019
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: C++
- Commits: 462
- Committers: 4
- Issues: 104
- Pull Requests: 25
- Owner: AICAN-Research
- Stars: 116
- Forks: 23
- Packages: 2

https://github.com/kyegomez/midas
Implementation of Midas from [Towards Robust Monocular Depth Estimation] in Pytorch and Zeta
ai
artificial-intelligence
ml
multi-modal
parallel
python
pytorch
tensorflow
vision-models
Added: over 1 year ago - Last Synced: 11 months ago
- Created: January 26, 2024

https://github.com/stephenapx/ucd-scm
[IGARSS 2024] Segment Change Model (SCM) for Unsupervised Change detection in VHR Remote Sensing Images: a Case Study of Buildings
change-detection
clip
deep-learning
fastsam
pytorch
remote-sensing
sam
satellite-imagery
Added: over 1 year ago - Last Synced: 11 months ago
- Created: December 21, 2023
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 3
- Committers: 2
- Issues: 2
- Pull Requests: 0
- Owner: StephenApX
- Stars: 27
- Forks: 1
- Packages: 0

https://github.com/pistony/torch-toolbox
π Toolbox to extend PyTorch functionalities
arcloss
autoaugment
cosinewarmuplr
cosloss
cutout
cv2
flops
initializers
l2softmax
labelsmoothing
lmdb
lmdb-dataset
lookahead
metrics
mixup
pytorch
swish
switchnorm
teansform
toolbox
Added: over 1 year ago - Last Synced: 11 months ago
- Created: May 06, 2019

https://github.com/openvinotoolkit/nncf
Neural Network Compression Framework for enhanced OpenVINOβ’ inference
bert
classification
compression
deep-learning
hawq
mixed-precision-training
mmdetection
nlp
object-detection
onnx
openvino
pruning
pytorch
quantization
quantization-aware-training
semantic-segmentation
sparsity
tensorflow
transformers
Added: over 1 year ago - Last Synced: 11 months ago
- Created: May 13, 2020
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 1778
- Committers: 67
- Issues: 61
- Pull Requests: 797
- Owner: openvinotoolkit
- Stars: 772
- Forks: 197
- Packages: 2
- Downloads: 71,044

https://github.com/muhd-umer/comyx
Comyx is an optimized and modular Python library for simulating wireless communication systems
6g
noma
numba
numpy
pytorch
ray
reinforcement-learning
ris
scipy
simulation
star-ris
wireless-communication
Added: over 1 year ago - Last Synced: 11 months ago
- Created: May 11, 2023

https://github.com/csailvision/semantic-segmentation-pytorch
Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
ade20k
pytorch
scene-recognition
semantic-segmentation
Added: over 1 year ago - Last Synced: 11 months ago
- Created: October 07, 2017
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 237
- Committers: 13
- Issues: 88
- Pull Requests: 15
- Owner: CSAILVision
- Stars: 4830
- Forks: 1080
- Packages: 1
- Downloads: 144

https://github.com/dsgoficial/pytorch_segmentation_models_trainer
Framework to train semantic segmentation models on Pytorch using yaml config files
hydra
pytorch
pytorch-lightning
semantic-segmentation
Added: over 1 year ago - Last Synced: 11 months ago
- Created: February 22, 2021
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 589
- Committers: 4
- Issues: 2
- Pull Requests: 98
- Owner: dsgoficial
- Stars: 15
- Forks: 1
- Packages: 0

https://github.com/kyegomez/soundstream
Implementation of SoundtStream from the paper: "SoundStream: An End-to-End Neural Audio Codec"
ai
artificial-intelligence
audio-model
audioai
codex
convolution
convolutional-layers
convolutional-neural-networks
machine-learning
ml
pytorch
transformer
Added: over 1 year ago - Last Synced: 11 months ago
- Created: January 04, 2024

https://github.com/kevinmusgrave/powerful-benchmarker
A library for ML benchmarking. It's powerful.
benchmarking
computer-vision
deep-learning
domain-adaptation
machine-learning
metric-learning
pytorch
transfer-learning
Added: over 1 year ago - Last Synced: 11 months ago
- Created: November 09, 2019
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Jupyter Notebook
- Commits: 861
- Committers: 3
- Issues: 86
- Pull Requests: 15
- Owner: KevinMusgrave
- Stars: 426
- Forks: 44
- Packages: 1
- Downloads: 198

https://github.com/pytorch/torcharrow
High performance model preprocessing library on PyTorch
preprocessing
python
pytorch
Added: over 1 year ago - Last Synced: 11 months ago
- Created: September 27, 2021

https://github.com/kyegomez/ast
Implementation of AST from the paper: "AST: Audio Spectrogram Transformer' in PyTorch and Zeta
ai
artificial-intelligence
attention
machine
machine-learning
machine-learning-algorithms
pytorch
pytorch-implementation
tensorflow
Added: over 1 year ago - Last Synced: 11 months ago
- Created: December 30, 2023

https://github.com/opendilab/lightzero
[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios
alpha-beta-pruning
alphazero
atari
board-game
board-games
continuous-control
efficientzero
gomoku
gumbel-muzero
gym
mcts
mcts-algorithm
monte-carlo-tree-search
muzero
pytorch
reinforcement-learning
sampled-muzero
self-play
stochastic-muzero
tictactoe
Added: over 1 year ago - Last Synced: 11 months ago
- Created: October 08, 2022

https://github.com/open-mmlab/mmocr
OpenMMLab Text Detection, Recognition and Understanding Toolbox
abcnet
abinet
crnn
dbnet
deep-learning
fcenet
key-information-extraction
maskrcnn
ocr
pan
panet
psenet
pytorch
sar
sdmg-r
segmentation-based-text-recognition
spts
svtr
text-detection
text-recognition
Added: over 1 year ago - Last Synced: 11 months ago
- Created: April 07, 2021
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 1107
- Committers: 93
- Issues: 135
- Pull Requests: 98
- Owner: open-mmlab
- Stars: 4131
- Forks: 726
- Packages: 2
- Downloads: 6,070

https://github.com/nvidia/dali
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
audio-processing
data-augmentation
data-processing
deep-learning
fast-data-pipeline
gpu
gpu-tensorflow
image-augmentation
image-processing
machine-learning
mxnet
neural-network
paddle
python
pytorch
Added: over 1 year ago - Last Synced: 11 months ago
- Created: June 01, 2018

https://github.com/n3slami/ef-raft
Ef-RAFT: Rethinking RAFT for Efficient Optical Flow (MVIP 2024)
computer-vision
cuda
optical-flow
pytorch
Added: over 1 year ago - Last Synced: 11 months ago
- Created: February 14, 2023

https://github.com/yihong1120/yolov8-license-plate-insights
This repository demonstrates YOLOv8-based license plate recognition with GCP Vision AI integration, enabling versatile real-world applications like vehicle identification, traffic monitoring, and geospatial analysis while capturing vital media metadata for enhanced insights.
computer-vision
data-augmentation
deep-learning
gcp
geospatial-analysis
license-plate-recognition
machine-learning
media-metadata
ml
object-detection
ocr
opencv
optical-character-recognition
python
pytorch
traffic-monitoring
ultralytics
vehicle-identification
vision-ai
yolov8
Added: over 1 year ago - Last Synced: 11 months ago
- Created: November 02, 2023
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Jupyter Notebook
- Commits: 59
- Committers: 2
- Issues: 2
- Pull Requests: 0
- Owner: yihong1120
- Stars: 3
- Forks: 1
- Packages: 0

https://github.com/wenjiedu/pypots
A Python toolkit/library for reality-centric machine/deep learning and data mining on partially-observed time series, including SOTA neural network models for scientific analysis tasks of imputation, classification, clustering, forecasting, & anomaly detection on incomplete industrial (irregularly-sampled) multivariate TS with NaN missing values
classification
clustering
data-mining
data-science
deep-learning
forecasting
healthcare
imputation
incomplete
industrial
interpolation
machine-learning
missing-values
missingness
neural-network
partially-observed-time-series
pytorch
science-research
time-series
time-series-analysis
Added: over 1 year ago - Last Synced: 11 months ago
- Created: March 29, 2022

https://github.com/swanhubx/swanlab
π§SwanLab: track and visualize all the pieces of your machine learning pipeline. θ·θΈͺδΈε―θ§εδ½ ηζΊε¨ε¦δΉ ε
¨ζ΅η¨
data-science
deep-learning
fastapi
jax
machine-learning
mlops
model-versioning
python
pytorch
tensorboard
tensorflow
tracking
transformers
visualization
Added: over 1 year ago - Last Synced: 11 months ago
- Created: November 24, 2023

https://github.com/magis-slac/gradoptics
differentiable
machine-learning
optics-simulation
pytorch
ray-tracing
simulator
tomography
Added: over 1 year ago - Last Synced: over 1 year ago
- Created: August 06, 2021
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 197
- Committers: 7
- Issues: 1
- Pull Requests: 11
- Owner: magis-slac
- Stars: 8
- Forks: 2
- Packages: 0

https://github.com/deeprank/deeprank2
An open-source deep learning framework for data mining of protein-protein interfaces or single-residue variants.
convolutional-neural-networks
deep-learning
graph-neural-networks
protein-protein-interfaces
pytorch
single-residue-variants
structural-biology
Added: over 1 year ago - Last Synced: 11 months ago
- Created: January 21, 2022

https://github.com/BiaPyX/BiaPy
Open source Python library for building bioimage analysis pipelines
biomedical-image-processing
classification
computer-vision
convolutional-neural-networks
deep-learning
denoising
image-processing
image-segmentation
instance-segmentation
machine-learning
medical-imaging
object-detection
python
pytorch
segmentation
self-supervised-learning
semantic-segmentation
super-resolution
Added: over 1 year ago - Last Synced: 11 months ago
- Created: February 28, 2019
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Jupyter Notebook
- Commits: 1308
- Committers: 15
- Issues: 25
- Pull Requests: 12
- Owner: BiaPyX
- Stars: 113
- Forks: 27
- Packages: 1
- Downloads: 370

https://github.com/mberkay0/pretrained-backbones-unet
A PyTorch-based Python library with UNet architecture and multiple backbones for Image Semantic Segmentation.
backbone-models
convnext
convnext-unet
densenet
efficientnet
pip
pretrained-models
pytorch
regnet
resnet
semantic-segmentation
timm
unet
vgg
Added: over 1 year ago - Last Synced: 11 months ago
- Created: March 10, 2023

https://github.com/meetps/pytorch-semseg
Semantic Segmentation Architectures Implemented in PyTorch
deep-learning
fully-convolutional-networks
pytorch
semantic-segmentation
Added: over 1 year ago - Last Synced: 11 months ago
- Created: March 22, 2017

https://github.com/volume-em/empanada
Panoptic segmentation algorithms for 2D and 3D electron microscopy images
deep-learning
electron-microscopy
instance-segmentation
panoptic-segmentation
pytorch
segmentation
semantic-segmentation
Added: over 1 year ago - Last Synced: 11 months ago
- Created: December 01, 2021

https://github.com/pycroscopy/atomai
Deep and Machine Learning for Microscopy
deep-kernel-learning
deep-learning
electron-microscopy
ensemble-learning
fully-convolutional-networks
google-colaboratory
imaging
jupyter-notebook
machine-learning
materials
materials-science
microscopy
multivariate-analysis
pytorch
scanning-probe-microscopy
semantic-segmentation
variational-autoencoders
Added: over 1 year ago - Last Synced: 11 months ago
- Created: September 04, 2020
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 1262
- Committers: 8
- Issues: 14
- Pull Requests: 69
- Owner: pycroscopy
- Stars: 181
- Forks: 37
- Packages: 2
- Downloads: 530

https://github.com/yoshitomo-matsubara/sc2-benchmark
[TMLR] "SC2 Benchmark: Supervised Compression for Split Computing"
coco
image-classification
imagenet
knowledge-distillation
object-detection
pascal-voc
pytorch
semantic-segmentation
split-computing
supervised-compression
Added: over 1 year ago - Last Synced: 11 months ago
- Created: October 24, 2021
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 246
- Committers: 4
- Issues: 2
- Pull Requests: 119
- Owner: yoshitomo-matsubara
- Stars: 23
- Forks: 7
- Packages: 1
- Downloads: 33

https://github.com/rwightman/efficientdet-pytorch
A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
efficientdet
efficientnet
object-detection
pytorch
semantic-segmentation
Added: over 1 year ago - Last Synced: 11 months ago
- Created: March 26, 2020

https://github.com/pytorch/rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
ai
control
decision-making
distributed-computing
machine-learning
marl
model-based-reinforcement-learning
multi-agent-reinforcement-learning
pytorch
reinforcement-learning
rl
robotics
torch
Added: over 1 year ago - Last Synced: 11 months ago
- Created: February 01, 2022

https://github.com/ami-iit/adam
adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.
adam-robotics
automatic-differentiation
casadi
jax
motion-planning
numpy
optimization
python
pytorch
reinforcement-learning
rigid-body-dynamics
robotics
urdf
Added: over 1 year ago - Last Synced: 11 months ago
- Created: May 25, 2021

https://github.com/helmholtz-analytics/heat/
Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
array-api
data-analytics
data-processing
data-science
distributed
gpu
hpc
machine-learning
massive-datasets
mpi
mpi4py
multi-gpu
multi-node-cluster
numpy
parallelism
python
pytorch
tensors
Added: over 1 year ago - Last Synced: 11 months ago
- Created: May 17, 2018
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 4590
- Committers: 62
- Issues: 218
- Pull Requests: 354
- Owner: helmholtz-analytics
- Stars: 193
- Forks: 56
- Packages: 1

https://github.com/kirill-vish/beyond-inet
Code for experiments for "ConvNet vs Transformer, Supervised vs CLIP: Beyond ImageNet Accuracy"
calibration
clip
computer-vision
convnet
deep-learning
evaluation
imagenet
invariant-features
model-mistake
pytorch
robustness
supervised
synthetic-data
transfer-learning
transformer
Added: over 1 year ago - Last Synced: 11 months ago
- Created: November 15, 2023
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 24
- Committers: 3
- Issues: 0
- Pull Requests: 1
- Owner: kirill-vish
- Stars: 76
- Forks: 5
- Packages: 0

https://github.com/intel/rho-diffusion
Parameterizing n-dimensional density fields using denoising diffusion probabilistic models
artificial-intelligence
density-estimation
diffusion-models
generative-model
hpc
physics
pytorch
Added: over 1 year ago - Last Synced: 11 months ago
- Created: November 22, 2023

https://github.com/honghanhh/fsdl_2022_solution
Solution of Full Stack Deep Learning - Course 2022
cnn
continual-learning
data-annotation
data-management
deep-learning
deployment
ethics
fullstack
pytorch
transformers
Added: over 1 year ago - Last Synced: 11 months ago
- Created: October 24, 2022
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: /projects/review
- Commits: 8
- Committers: 1
- Issues: 1
- Pull Requests: 0
- Owner: honghanhh
- Stars: 2
- Forks: 1
- Packages: 0

https://github.com/ignf/odeon
ODEON is a task-agnostic framework for deep learning applied to remote sensing
deep-learning
earth-observation
gis
pytorch
pytorch-lightning
rasterio
remote-sensing
Added: over 1 year ago - Last Synced: 11 months ago
- Created: June 23, 2022

https://github.com/zeqiang-lai/dphsir
PyTorch implementation of "Deep Plug-and-Play Prior for Hyperspectral Image Restoration" (Neurocomputing 2022)
hyperspectral
image-restoration
plug-and-play
pytorch
Added: over 1 year ago - Last Synced: 11 months ago
- Created: December 14, 2021
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 24
- Committers: 2
- Issues: 3
- Pull Requests: 0
- Owner: Zeqiang-Lai
- Stars: 30
- Forks: 12
- Packages: 0

https://github.com/paccmann/paccmann_datasets
pytoda - PaccMann PyTorch Dataset Classes. Read the docs: https://paccmann.github.io/paccmann_datasets/
bioinformatics
chemoinformatics
deep-learning
python
pytorch
rdkit
smiles
Added: over 1 year ago - Last Synced: 11 months ago
- Created: November 01, 2019

https://github.com/mv1388/aitoolbox
PyTorch Model Training and Experiment Tracking Framework
deep-learning
machine-learning
nlp
python
pytorch
research
Added: over 1 year ago - Last Synced: 11 months ago
- Created: July 18, 2017

https://github.com/allenai/smashed
SMASHED is a toolkit designed to apply transformations to samples in datasets, such as fields extraction, tokenization, prompting, batching, and more. Supports datasets from Huggingface, torchdata iterables, or simple lists of dictionaries.
dataset
datasets
dict
huggingface
in-context-learning
mappers
natural-language-processing
nlp
pipeline
prefix
prefix-tuning
preprocessing
prompting
pytorch
text
torchdata
transformer
transformers
Added: over 1 year ago - Last Synced: 11 months ago
- Created: July 21, 2022
