‘’‘einops有三个常用方法:rearrange,repeat,reduce’‘’
rearrange的操作相当于转置
rearrange(image,‘h w c -> w h c’) 高和宽转置
path = '../data/cat_and_mouse.jpg'
image = cv2.imread(path)
h,w,c = image.shape # shape第一个值是h,第二个是w
image = cv2.resize(image,(w*2,h*2)) #resize处(w,h)
print(image.shape)
cv2.imshow('image',image)
cv2.waitKey()
cv2.destroyAllWindows()
原图(h,w,c) = (281,500,3) 为了后面将图片一分为2的操作,把图片尺寸放大两倍:
(h2,w2,c) = (562,1000,3)
# 1.变换 h,w,c的位置: 如把h,w,c 变成 w,h,c
image_wh = rearrange(image,'h w c -> w h c')
print(image_wh.shape)
cv2.imshow('image_wh',image_wh)
cv2.imwrite('../data/cat_and_mouse_new.jpg',image_wh)
cv2.waitKey()
cv2.destroyAllWindows()
(h,w,c) = (1000,562,3)
image = rearrange(image,‘h (a w) c -> w (a h) c’,a=2) 是将宽一分为2,变成两个图片,然后两个图片分别转置
h (2 w) c = (562,2x500,3)是把图片的按宽的方向一分为2,变成两个图片。w原来是1000,(2 w) 将w变成500。
将一分为2的两个图片转置。由(h,w,c) = (562,2x500,3)变成 w, (2 h) c = (500,562,3) 拼接 (500,562,3) = (500,562x2,3)
repeat操作
repeat是指定张量在哪个维度上复制。
image = repeat(image, 'h w c -> h (repeat w) c', repeat=2)
如在宽上复制:
reduce操作
详见:
https://www.cnblogs.com/c-chenbin/p/15375637.html