search

JDSobek / MedYOLO

A 3D bounding box detection model for medical data.
GNU Affero General Public License v3.0
36 stars 9 forks source link

About the label. #27

Open mawei-north opened 5 days ago

mawei-north commented 5 days ago

May I ask how the label file for your dataset is (.txt) generated or made? What software was used to achieve this?

JDSobek commented 3 days ago

I use this script to convert semantic segmentations into the bounding box txt labels for MedYOLO. It's a fairly simple script though (it only looks for 1 bounding box for each class), and for some datasets, like LIDC, the instance segmentations available in the dataset allow you to do some more sophisticated labeling than my script can do. I generated my segmentations with RIL-Contour, but any nifti segmentation masks will work.

I'd recommend using my script with a grain of salt, in case your dataset or your goal with the model requires more complex processing than mine did.