local variable 'truth_bboxes' referenced before assignment

uci-cbcl / NoduleNet

[MICCAI' 19] NoduleNet: Decoupled False Positive Reduction for Pulmonary Nodule Detection and Segmentation

Other

185 stars 57 forks source link

Open jrmathew1 opened 2 years ago

jrmathew1 commented 2 years ago

I am training the model on a new dataset and got this error during epoch 6:

File "/data2/NoduleNet/dataset/mask_reader.py", line 124, in getitem

return [torch.from_numpy(input).float(), truth_bboxes, truth_labels, truth_masks, masks]

UnboundLocalError: local variable 'truth_bboxes' referenced before assignment