Metadata-Version: 2.1
Name: segmentation-models-pytorch-3d
Version: 1.0.1
Summary: Set of models for segmentation of 3D volumes using PyTorch.
Home-page: https://github.com/ZFTurbo/segmentation_models_pytorch_3d
Author: Roman Sol (ZFTurbo)
Author-email: UNKNOWN
License: UNKNOWN
Platform: UNKNOWN
Requires-Dist: torch
Requires-Dist: torchvision (>=0.5.0)
Requires-Dist: pretrainedmodels (==0.7.4)
Requires-Dist: efficientnet-pytorch (==0.7.1)
Requires-Dist: timm (==0.9.7)
Requires-Dist: tqdm
Requires-Dist: pillow
Requires-Dist: six

3D variants of popular models for segmentation like FPN, Unet, Linknet etc using Pytorch module.Automatic conversion of 2D imagenet weights to 3D variant


