Image are displayed as white

[nik1806]

Single-Human-Keypoint-Detection/utils/LIPDataset.py

Line 72 in 06e7fdb

return {'image': torch.from_numpy(image).type(torch.FloatTensor),

After the above changes, the untransform_n_display function is having issue displaying samples. The images appear as a white background, whereas the marking is intact.