https://github.com/eidoslab/torch-segmentation-3d
https://github.com/eidoslab/torch-segmentation-3d
Last synced: 7 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/eidoslab/torch-segmentation-3d
- Owner: EIDOSLAB
- Created: 2022-07-12T09:18:08.000Z (over 3 years ago)
- Default Branch: master
- Last Pushed: 2024-04-07T13:48:17.000Z (almost 2 years ago)
- Last Synced: 2024-04-07T16:40:42.021Z (almost 2 years ago)
- Language: Python
- Size: 85.9 KB
- Stars: 0
- Watchers: 6
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
3D Segmentations models for pytorch
Supported architectures:
- Unet
- Unet++
Install:
```bash
git clone git@github.com:EIDOSLAB/torch-segmentation-3d.git && cd torch-segmentation-3d
pip3 install .
```
Example:
```python
import torch_segmentation_3d as seg3d
model = seg3d.models.Unet(encoder_name="resnet18", encoder_weights="imagenet", in_channels=1, classes=3)
```
Supported encoders:
- ResNet
| Encoder | Weights | Params |
|----------|----------|---------|
| resnet18 | imagenet | 33M |
| resnet34 | imagenet | 63M |
| resnet50 | imagenet | 46M |
| resnet101 | imagenet | 85M |
Cache location:
- Linux: `~/.cache/torch/hub/checkpoints/resnet18-78569d5e.pth` (imagenet-1k)