Python代码 一行实现 NHWC BGR格式图片 转 NCHW RGB格式图片。
高效、简洁的python代码~,非常好,就是看的时候一眼扫过 容易忽略。。。
# Read image
img = cv2.imread(img_path) # BGR
# nhwc bgr -> nchw rgb
img = img[:, :, ::-1].transpose(2, 0, 1)
Python代码 一行实现 NHWC BGR格式图片 转 NCHW RGB格式图片。
高效、简洁的python代码~,非常好,就是看的时候一眼扫过 容易忽略。。。
# Read image
img = cv2.imread(img_path) # BGR
# nhwc bgr -> nchw rgb
img = img[:, :, ::-1].transpose(2, 0, 1)