PixelLink-with-pytorch
PixelLink-with-pytorch copied to clipboard
A tiny change
In datasets.py line 351: "label = sample['label'].reshape([-1, 4, 2])", should be changed to "label = np.array(sample['label']['coor']).reshape([-1, 4, 2])".
Because sample['label'] is dict type, and it only need coordinates to draw the contours.