d5ca4b4b83a4d759ba81737e0bb12cdd591932d3,codes/test_seg.py,,,#,17

Before Change

    // prob
    torch.save(output, os.path.join(save_prob_path, base+"_bic.pth"))  // 1x8xHxW
    // byte img
    output = output.squeeze()
    _, argmax = torch.max(output, 0)
    argmax = argmax.squeeze().byte()
    cv2.imwrite(os.path.join(save_byteimg_path, base+".png"), argmax.numpy())
    // color img

After Change

])).float()
lookup_table /= 255

print("seg testing...")

idx = 0
for path in glob.glob(test_img_folder + "/*"):
    idx += 1

Instances