Skip to main content

unet3d

pip install unet3d to get single-file implementations of 3D UNet architectures for 3D image segmentation!

Usage 💡

Run a UNet3d...

import torch
from unet3d import UNet3d, LinkNet3d

unet = UNet3d(c_in=1, c_out=2, c_start=8, layers=4)
x = torch.rand(1, 1, 64, 64, 64)
output = unet(x)
print(output.shape)  # (1, 2, 64, 64, 64)

...or a LinkNet3d—a more compute efficient UNet variant—on torch.random data 🤓

linknet = LinkNet3d(c_in=1, c_out=2, c_start=8, layers=4)
x = torch.rand(1, 1, 64, 64, 64)
output = linknet(x)
print(output.shape)  # (1, 2, 64, 64, 64)

Release files for unet3d 0.1.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for unet3d 0.1.0
File Size Uploaded
unet3d-0.1.0.tar.gz 2.5 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for unet3d 0.1.0
File Interpreter ABI Platform
unet3d-0.1.0-py3-none-any.whl Python 3 none any Details

Total release size: 6.0 kB

Release files / unet3d-0.1.0.tar.gz

Download URL unet3d-0.1.0.tar.gz
Size 2.5 kB
Tags Source
SHA-256 checksum
How to use checksums
9a76e15ce0fdc7926fd1caa315eab94b793df56a26c46ad671d17d0ff0c66319
BLAKE2b-256 checksum
How to use checksums
6fcfcdfd63b83fb9a25d5d20cc8c6b23b4eb32271b5634aba2ff16e00e4e55e3
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via poetry/2.0.1 CPython/3.12.7 Linux/5.15.0-56-generic

Release files / unet3d-0.1.0-py3-none-any.whl

Download URL unet3d-0.1.0-py3-none-any.whl
Size 3.6 kB
Tags Python 3
SHA-256 checksum
How to use checksums
c89c7504ff695e5011cb100416fe1979a512f750388a4663f9d86d3a52f61aeb
BLAKE2b-256 checksum
How to use checksums
0b4d75c20b01e1523dcf193ecd7f7be16255587edc6d2fea82ddc2329c446434
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via poetry/2.0.1 CPython/3.12.7 Linux/5.15.0-56-generic

Release history Release notifications | RSS feed

This release

0.1.0 This release

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page