PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
qubvel-org
segmentation-models-pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

704K 12K 2K
qubvel
segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

19K 5K 1K
fepegar
unet

"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.

6K 198 24
yingkaisha
keras-unet-collection

The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.

4K 759 185
qubvel
d3m-segmentation-models-pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

3K 12K 2K
qubvel
segmentation-models-pytorch-deepflash2

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

3K 12K 2K
archinetai
a-unet

A toolbox that provides hackable building blocks for generic 1D/2D/3D UNets, in PyTorch.

3K 89 9
CAIIVS
chuchichaestli

Where you find all the state-of-the-art cooking utensils (salt, pepper, gradient descent... the usual).

2K 5 1
org-arl
unetpy

UnetSocket: A high-level socket interface for UnetStack in Python

1K 2 4
desmoteo
swiss-army-keras

A library to help with the development of AI models with Keras, with a focus on edge / IoT applications. Based originally on https://github.com/yingkaisha/keras-unet-collection

979 0 0
org-arl
arlpy

ARL Python Tools

771 146 44
lucidrains
metnet3-pytorch

MetNet 3 - Pytorch

722 240 28
karolzak
keras-unet

Helper package with multiple U-Net implementations in Keras as well as useful utility tools helpful when working with image segmentation tasks

588 342 110
lucidrains
uformer-pytorch

Implementation of Uformer, Attention-based Unet, in Pytorch

494 96 16
IanTaehoonYoo
seg-torch

Semantic Segmentation with Pytorch

458 88 20
will-rice
denoisers

A package for training audio denoisers

408 41 2
zhoudaxia233
efficientunet-pytorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

364 161 34
nemo256
bc-count

Count red, white blood cells to detect various diseases such as blood cancer (leukemia), lower red blood cells (anemia)...

354 8 0
thenewflesh
flatiron

Standardized computer vision pipeline framework with a chainable API and procedurally defined model architecture

335 2 0
sovit-123
torch-segment

A library for deep learning image segmentation using PyTorch

202 0 0
zhoudaxia233
efficientunet

Keras Implementation of Unet with EfficientNet as encoder.

198 48 26
qubvel
segmentation-models-zwy

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

193 5K 1K
samsonq
covidxai

Implementation of explainability algorithms (layer-wise relevance propagation, local interpretable model-agnostic explanations, gradient-weighted class activation mapping) on computer vision architectures to identify and explain regions of COVID 19 pneumonia in chest X-ray and CT scans.

191 6 0
wolny
pytorch3dunet

3D U-Net model for volumetric semantic segmentation written in pytorch

190 2K 562
    • Data from PyPI, GitHub, ClickHouse, and BigQuery