KeyError when trying to train in toy_exp

[17alia]

I am trying to train using the donuts_shape dataset in the toy_exp and I am getting the following error:

How can I fix this? it looks like the dictionary "batch" has keys 'data', 'seg', 'pid', 'class_target', 'bb_target', and 'roi_masks', but not 'roi_labels'.

Thanks for any help you can provide.

[themantalope]

I'm also having a similar issue.

[themantalope]

I think this is an issue with the batchgenerators repository. The medicaldetectiontoolkit repository expects batchgenerators to be version 0.19.3. If you install the latest version, the output of the generator doesn't include the 'roi_labels' key. Might be worth submitting an issue over there.

[themantalope]

@17alia

pip uninstall batchgenerators
pip install batchgenerators==0.19.3