https://github.com/gregspangenberg/armcrop

A machine learning model that does oriented bounding box cropping of CT scans to any bone of interest in the arm
https://github.com/gregspangenberg/armcrop

Last synced: about 1 year ago
JSON representation

A machine learning model that does oriented bounding box cropping of CT scans to any bone of interest in the arm

• Host: GitHub
• URL: https://github.com/gregspangenberg/armcrop
• Owner: gregspangenberg
• License: mit
• Created: 2024-09-26T14:44:44.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2025-03-13T04:39:08.000Z (over 1 year ago)
• Last Synced: 2025-03-13T05:27:50.282Z (over 1 year ago)
• Language: Python
• Homepage:
• Size: 829 KB
• Stars: 1
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: license.md

Awesome Lists containing this project

README

          
# *armcrop*

[![PyPI Latest Release](https://img.shields.io/pypi/v/armcrop.svg)](https://pypi.org/project/armcrop)

[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)

This package contains a machine learning model that can crop a CT scan to any of the following classes:

- clavicle

- scapula

- humerus

- radius_ulna

- hand

It can do both axis aligned cropping and oriented bounding box (obb) cropping. The obb cropping allows for rigid registration of CT scans.

This model was trained on arm only scans and proximal half chest scans. A limitation of this model is that when using a full chest CT where the spine is present the model will struggle to differentiate between it and the scapula.

## Installation

To use CPU execution install with:

```

pip install armcrop

```

To use CUDA execution you will need to also install torch and onnxruntime-gpu.