PixelLink-with-pytorch

PixelLink-with-pytorch copied to clipboard

postprocess.py line 89:128 should be changed to 256.

zhs108

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...

ReadyH