Open 17alia opened 3 years ago
I'm also having a similar issue.
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.
@17alia
pip uninstall batchgenerators
pip install batchgenerators==0.19.3
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.