opencv读写文件操作

1. 读取图像文件

   • OpenCV 中使用cv2.imread()函数来读取图像文件。这个函数的第一个参数是图像文件的路径，第二个参数是一个可选的标志，用于指定读取图像的方式。
   • 示例代码：

   import cv2

   以彩色模式读取图像（这是默认模式）

   img = cv2.imread('your_image.jpg', cv2.IMREAD_COLOR)
   if img is None:
   print("无法读取图像")
   else:
   print("图像读取成功，形状为:", img.shape)

• 在这个示例中，cv2.IMREAD_COLOR（值为 1）表示以彩色模式读取图像。如果要以灰度模式读取，可以使用cv2.IMREAD_GRAYSCALE（值为 0），例如img = cv2.imread('your_image.jpg', cv2.IMREAD_GRAYSCALE)。如果想读取包含 alpha 通道的图像，可以使用cv2.IMREAD_UNCHANGED（值为 - 1）。

1. 读取视频文件

   • 对于视频文件，使用cv2.VideoCapture()函数。它可以打开视频文件或者摄像头设备。
   • 示例代码：

   import cv2

   打开视频文件

   cap = cv2.VideoCapture('your_video.mp4')
   if not cap.isOpened():
   print("无法打开视频文件")
   else:
   while True:
   ret, frame = cap.read()
   if not ret:
   break
   # 在这里对每一帧进行处理，例如显示
   cv2.imshow('Video', frame)
   if cv2.waitKey(25) & 0xFF == ord('q'):
   break
   cap.release()
   cv2.destroyAllWindows()

• 在这个示例中，cap.read()函数返回两个值。ret是一个布尔值，表示是否成功读取帧，frame是读取到的视频帧。cv2.waitKey()函数用于等待用户按键，参数是等待的时间（以毫秒为单位），这里等待 25 毫秒。如果用户按下q键，就退出循环。最后，cap.release()释放视频文件资源，cv2.destroyAllWindows()关闭所有打开的窗口。

1. 写入图像文件

   • 使用cv2.imwrite()函数来保存图像。它的第一个参数是要保存的文件路径和文件名，第二个参数是要保存的图像数据。
   • 示例代码：

   import cv2
   img = cv2.imread('input_image.jpg', cv2.IMREAD_COLOR)

   进行一些图像处理，例如转换为灰度图

   gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

   保存处理后的图像

   cv2.imwrite('output_gray_image.jpg', gray_img)

• 在这个示例中，首先读取了一张彩色图像，然后使用cv2.cvtColor()函数将其转换为灰度图，最后将灰度图保存为一个新的文件。

1. 写入视频文件

   • 要写入视频文件，需要先创建一个cv2.VideoWriter对象。这个对象的构造函数需要指定输出视频文件的路径、视频编解码器、帧率和视频尺寸等参数。
   • 示例代码：

   import cv2

   读取视频文件

   cap = cv2.VideoCapture('input_video.mp4')
   if not cap.isOpened():
   print("无法打开输入视频文件")
   else:
   # 获取视频的帧率、尺寸等信息
   fps = cap.get(cv2.CAP_PROP_FPS)
   width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
   height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
   # 创建VideoWriter对象，使用MJPG编解码器
   fourcc = cv2.VideoWriter_fourcc(*'MJPG')
   out = cv2.VideoWriter('output_video.avi', fourcc, fps, (width, height))
   while True:
   ret, frame = cap.read()
   if not ret:
   break
   # 在这里可以对帧进行处理，例如添加文字或绘制图形
   out.write(frame)
   cap.release()
   out.release()
   cv2.destroyAllWindows()

• 在这个示例中，首先打开一个输入视频文件，获取其帧率、宽度和高度等信息。然后使用cv2.VideoWriter_fourcc()函数指定视频编解码器（这里是 MJPG），创建一个cv2.VideoWriter对象来保存处理后的视频。在循环中，读取输入视频的每一帧，进行处理（这里只是简单地将原始帧写入输出视频），最后释放资源。

请注意，在实际使用中，你需要将代码中的your_image.jpg、your_video.mp4等文件路径替换为真实的文件路径。而且，在写入视频文件时，编解码器的选择要根据你的系统和需求来确定，不同的编解码器可能在不同的系统上有不同的支持情况。