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

Fix missing disc 1 issue #6 #7

Closed NathanMolinier closed 1 year ago

NathanMolinier commented 1 year ago

Description

This PR fixes an issue where disc 1 was removed from the ground truth even when present in the original data, therefore making the hourglass unable to identify this disc.

The error came from the function mask2label where a condition prevented this label to be kept.

Moreover, due to its high complexity, the function was rewritten using the method getNonZeroCoordinates available with the Image object from spinalcordtoolbox

Fixed issue

issue https://github.com/spinalcordtoolbox/disc-labeling-hourglass/issues/6