和常用的图像处理库不一样的是，opencv读取的图像是bgr格式。所以在使用plt 或者PIL工具进行显示的时候，需要十分注意要进行颜色空间的转换。
import mathimport randomfrom typing import Anyfrom PIL import Imageimport cv2import numpy as npimport torchfrom torchvision.transforms import functional as Fimport matplotlib.pyplot as pltdef random_vertically_flip(image: np.ndarray, p: float = 0.5) -> np.ndarray:"""Flip an image horizontally randomlyArgs:image (np.ndarray): Image read with OpenCVp (optional, float): Vertically flip probability. Default: 0.5Returns:vertically_flip_image (np.ndarray): image after vertically flip"""if random.random() < p:vertically_flip_image = cv2.flip(image, 0)else:vertically_flip_image = imagereturn vertically_flip_imageif __name__=='__main__':image=Image.open('./data/test/Set5/baby.png')image.show(image)image=np.array(image)image=random_vertically_flip(image)hh=Image.fromarray(image)hh.show()# image=cv2.cvtColor(image,cv2.COLOR_RGB2BGR)# cv2.imshow('image',image)# cv2.waitKey(0)# plt.imshow(image)# plt.show()
