Skip to content

Add test to verify 'Lesion' label is unified to 'Lung Lesion'#178

Open
Tiago08 wants to merge 1 commit into
mlmed:mainfrom
Tiago08:fix/unify-lesion-labels
Open

Add test to verify 'Lesion' label is unified to 'Lung Lesion'#178
Tiago08 wants to merge 1 commit into
mlmed:mainfrom
Tiago08:fix/unify-lesion-labels

Conversation

@Tiago08

@Tiago08 Tiago08 commented Jun 15, 2026

Copy link
Copy Markdown

Closes #69

The issue noted that VinBrain uses "Lesion" while MIMIC uses "Lung Lesion" for the same pathology.

After investigating the codebase, the unification is already handled in datasets.py via np.char.replace on line 1358, and default_pathologies correctly uses "Lung Lesion".

This PR adds a regression test to test_dataloaders.py to ensure this consistency is maintained going forward.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Output labels "Lesion" and "Lung Lesion" exist in different datasets. They should be unified.

2 participants