PixelLink-with-pytorch icon indicating copy to clipboard operation
PixelLink-with-pytorch copied to clipboard

A tiny change

Open ReadyH opened this issue 7 years ago • 0 comments

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.

ReadyH avatar Oct 09 '18 08:10 ReadyH