search

spinalcordtoolbox / disc-labeling-hourglass

Labeling of intervertebral discs using the Hourglass deep learning architecture.
GNU Lesser General Public License v2.1
1 stars 0 forks source link

Extracting the number of the disc from the hourglass network #4

Closed NathanMolinier closed 1 year ago

NathanMolinier commented 1 year ago

Description

Currently, extracting discs numbers from the hourglass prediction is not possible, ground truth coordinates need to be used to estimate these numbers. Functions like closest_node in _test/testutils.py need to be used.

Solution

Each disc detection corresponds to a different class, therefore it should be possible to extract these numbers from the hourglass prediction.

NathanMolinier commented 1 year ago

This issue was fixed by a800e914b2ee9b5814494ce3a1273adc018e8783