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

Uploaded Python 3

File details

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

File metadata

  • Download URL: armcortnet-0.2.2.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.2.tar.gz
Algorithm Hash digest
SHA256 aa0a7260367b573aa9d6ecc835288a83e8dba4777956805022d054b9d119cd14
MD5 d786f81167e18b6a5e97f19da53c2c54
BLAKE2b-256 3192906edc57cf4435757ba1ee44870757adf44964f93ba7b347a79dcc7d73f9

See more details on using hashes here.

File details

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

File metadata

  • Download URL: armcortnet-0.2.2-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.2-py3-none-any.whl
Algorithm Hash digest
SHA256 828b845b9b7b9661478e76d7f4d271238b673ca6a20ff19c6d835b3bd68d2788
MD5 321c13aae4f2dbd50928f52f44b06251
BLAKE2b-256 6239fb11af54ea1dde58ce5d0bae74e0656fa307a8f2f9bc139830833cc0af4f

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