Open gazibrother opened 5 years ago
cv2读图片出来的shape是(H, W, C), 而PIL把图片读出来的shape是(W, H, C),所以您在data_loader这一块,这两种方式的混用,会影响到最后的loss计算吧
np.array会修正这个问题 会把图片转成 (H, W, C)
cv2读图片出来的shape是(H, W, C), 而PIL把图片读出来的shape是(W, H, C),所以您在data_loader这一块,这两种方式的混用,会影响到最后的loss计算吧