search

meteorshowers / RCF-pytorch

Richer Convolutional Features for Edge Detection model in pytorch CVPR2017
510 stars 145 forks source link

关于数据加载，因为cv2和PIL读出来的图片size不一致 #20

Open gazibrother opened 5 years ago

gazibrother commented 5 years ago

cv2读图片出来的shape是（H, W, C)， 而PIL把图片读出来的shape是(W, H, C），所以您在data_loader这一块，这两种方式的混用，会影响到最后的loss计算吧

gaieepo commented 5 years ago

np.array会修正这个问题 会把图片转成 (H, W, C)