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.3.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.3-py3-none-any.whl (4.2 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: armcortnet-0.2.3.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.3.tar.gz
Algorithm Hash digest
SHA256 a86b7f2bbf3815d5f37b32d4bb1e156a9b7b6bc120607bae8ee6a7ed6c1f0f4f
MD5 10b2d7c8cb49e2103c06cd6bc3538966
BLAKE2b-256 7f5877132c7b4fe8fc3a2aed421ab8d92eddf5950ab636c68b1d1c78b1e53e43

See more details on using hashes here.

File details

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

File metadata

  • Download URL: armcortnet-0.2.3-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.3-py3-none-any.whl
Algorithm Hash digest
SHA256 34aa67e33b3c5cec48ae2c50cfcc0232b82761a6c926160fb71553532e7df503
MD5 aab3cf444ffb45cce29f85833bcc43c7
BLAKE2b-256 bed79b5c43173620c3757c380a559bce57707dd93bf6336838c824f5e3e0a372

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