PiePline toolbelt
Installation:
pip install pietoolbelt
Install latest version before it's published on PyPi
pip install -U git+https://github.com/PiePline/pietoolbelt
Functional
-
augmentations
-
augmentations.detection
- predefined augmentations for detection task -
augmentations.segmentation
- predefined augmentations for segmentation task
-
-
datasets
-
datasets.stratification
- stratification by histogram -
datasets.utils
- set of datasets constructors that
-
-
losses
-
losses.common
- losses utils -
losses.regression
- regression losses -
losses.segmentation
- losses for single and multi-class segmentation -
losses.detection
- losses for detection task
-
-
metrics
--
metrics.common
- common utils for metrics -
cpu
- metrics, that calculates bynumpy
-
metrics.cpu.classification
- classification metrics -
metrics.cpu.detection
- detection metrics -
metrics.cpu.regression
- regression metrics -
metrics.cpu.segmentation
- segmentation metrics
-
-
torch
- metrics, that calculates bytorch
-
metrics.torch.classification
- classification metrics -
metrics.torch.detection
- detection metrics -
metrics.torch.regression
- regression metrics -
metrics.torch.segmentation
- segmentation metrics
-
-
-
models
- models zoo and constructors-
decoders.unet
- UNet decoder, that automatically constructs by encoder -
encoders.common
- basic interfaces for encoders -
encoders.inception
- InceptionV3 encoder -
encoders.mobile_net
- MobileNetV2 encoder -
encoders.resnet
- ResNet encoders -
albunet
- albunet model -
utils
- models utils -
weights_storage
- pretrained weights storage
-
-
steps
- some training process steps-
regression.train
- train step for regression task -
regression.bagging
- bagging step for regression task -
segmentation.bagging
- bagging step for segmentation task -
segmentation.inference
- inference for segmentation model -
segmentation.predict
- predict step for segmentation task -
stratification
- dataset stratification class
-
-
img_matcher
- image comparison and matching tool based on descriptors -
mask_composer
- mask composer tools that can effectively combine masks for regular, instance or multiclass segmentation -
train_config
- some predefined train configs for PiePline -
tta
- test time augmentations -
utils
- some utils -
viz
- image visualisation tools