Skip to main content

An automated deep learning pipeline for segmentation of the scapula, humerus, and their respective subregions in CT scans.

Project description

armcortnet

Armcortnet provides automatic segmentation of the humerus and scapula from CT scans. The deep learning model is trained to also segment out the cortical and trabecular subregions from each bone as well.

The deep learning pipeple consists of using armcrop to crop to an oriented bounding box around each humerus or scapula in the image and then a neural network based traine from the nnUNet framework segments that cropped volume. The segmetnation is then transformed back to the original coordinate system, post-processed and finally saved as a .seg.nrrd file.

Installation

pip install armcortnet

Usage

from armcortnet import Net

# Initialize segmentation model
model = Net(bone_type="scapula")  # or "humerus"

# Perform segmentation
model.predict(
    vol_path="path/to/input/ct.nrrd",
    output_seg_path="path/to/output/segmentation.seg.nrrd"
)

Output Labels

The segmentation output contains the following labels:

  • 0: Background
  • 1: Other adjacent bones ("i.e clavicle, radius, ulna, etc.")
  • 2: Cortical region of bone of interest
  • 3: Trabecular region of bone of interest

Models

Trained models are automatically downloaded from HuggingFace Hub (gregspangenberg/armcortnet) on first use.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

armcortnet-0.2.1.tar.gz (3.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

armcortnet-0.2.1-py3-none-any.whl (4.2 kB view details)

Uploaded Python 3

File details

Details for the file armcortnet-0.2.1.tar.gz.

File metadata

  • Download URL: armcortnet-0.2.1.tar.gz
  • Upload date:
  • Size: 3.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.0.1 CPython/3.12.3 Linux/6.8.0-53-generic

File hashes

Hashes for armcortnet-0.2.1.tar.gz
Algorithm Hash digest
SHA256 d1ef4502cc13e0a19b889a04b5b67bba65f969063550934262ca700dfac9f578
MD5 50a9efc4c8a9252fc5b19bd64612df22
BLAKE2b-256 47581c75f40f67f14283e32534c2a72478f7a266d8527214da3e1684d5a43a6d

See more details on using hashes here.

File details

Details for the file armcortnet-0.2.1-py3-none-any.whl.

File metadata

  • Download URL: armcortnet-0.2.1-py3-none-any.whl
  • Upload date:
  • Size: 4.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.0.1 CPython/3.12.3 Linux/6.8.0-53-generic

File hashes

Hashes for armcortnet-0.2.1-py3-none-any.whl
Algorithm Hash digest
SHA256 656beb66cb94f21654945b1d95bcb8f64ee50b2d0188f5d4d284e76accb0bd9e
MD5 d31aea90f212e150657861f048e97a20
BLAKE2b-256 b9eec53c2d6cf3698a3592bd0f3f25c4957009537bcbb9f29911b83536d69135

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page