search

intel / pyMIC

BSD 3-Clause "New" or "Revised" License
58 stars 16 forks source link

A problem occured when reading .nii.gz #23

Closed wuwo007 closed 3 years ago

wuwo007 commented 3 years ago

Dear authors, thank you for sharing this helpfull project. I have occured one problem when reading the DICOM CT volume (.nii.gz) and its labeled image(.nii.gz)。 In the train function of net_run_agent.py, when i run this sentence "labels_prob = self.convert_tensor_type(data['label_prob'])", the error "KeyError: 'label_prob'" appeared. I have checked the loaded training data, there is a attribute named "label", but not "label_prob". Could you please help me to solve this problem? Is there some requirements for the input labeled images? Thanks!

wuwo007 commented 3 years ago

Sorry for confusing a homonymic project