Bacth generation for LIDC dataset about medicaldetectiontoolkit

[pfjaeger]

hi, thank you for the feedback.
where in the code do you get these outputs?
class targets with values [0,1] are present before ConvertSegToBoundingBoxCoordinates, where values are modified via +=1 to make space for the 0=background class. So something is wrong in your data loader. Do you call the ConvertSegToBoundingBoxCoordinates method?

'I thought the class_target contains labels of the belonging of nodules to a class' - this is correct.
So in your output dict from load_dataset, batch elements with seg !=0 need to have filled class target lists, and empty segs need to have empty class target lists. otherwise your preprocessed data is incorrect.

Feel free to follow up on this comment and provide further info to figure this out.