search

OpenGVLab / SAM-Med2D

Official implementation of SAM-Med2D
Apache License 2.0
859 stars 80 forks source link

UnboundLocalError: local variable 'label' referenced before assignment #39

Open Marsyule opened 10 months ago

Marsyule commented 10 months ago

File "/root/autodl-tmp/SAM-Med2D-main/utils.py", line 110, in select_random_points labels.append(label) UnboundLocalError: local variable 'label' referenced before assignment

Marsyule commented 9 months ago

这个应该是utils.py中的def select_random_points(pr, gt, point_num = 9)中的bug，在这个函数应该修改的地方是
for i in selected_indices: x, y = i[0], i[1] if one_pred[x,y] == 0 and one_gt[x,y] == 1: label = 1 elif one_pred[x,y] == 1 and one_gt[x,y] == 0: label = 0 这个可能忽略了one_pred[x,y]==1、one_gt[x,y]==1和0、0的情况，我在后面加上了 elif one_pred[x,y] == 1 and one_gt[x,y] == 1: label = 1 else: label = 0 不知道对不对。