使用cv2对视频指定区域进行去噪

视频去噪其实和图象一样，只是需要现将视频截成图片，在对图片进行去噪，将去噪的图片在合成视频就行。可以利用cv2.imread()、imwrite()等轻松实现。

去噪步骤

1、视频逐帧读成图片

2、图片指定区域批量去噪

2、去噪后的图片写入视频

1、视频逐帧读成图片

有两种方法，第一种是直接将图片读入内存，图片就是以numpy格式存在内存中，优点是可以极快的增加图片的读取速度，但过长的视频可能会导致内存爆满。第二种是逐帧将图片读到本地文件夹，以png的格式存到指定路径中，优点是不占用内存，但是由于加入了cv2.imwrite()，会耗费一定的时间。由于我这次不在服务器上跑，所以选择第二种方法。

def make_frame_img(video_path, output_path):"""从视频文件中抽取帧数图片，保存到指定路径参数：video_path: 输入视频路径output_path: 保存帧图片文件夹return 无"""# 读取视频文件cap = cv2.VideoCapture(video_path)# 检查视频是否成功打开if not cap.isOpened():print("无法打开视频文件：", video_path)return# 获取视频帧率和总帧数fps = cap.get(cv2.CAP_PROP_FPS)total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))# 如果没有创建输出文件夹，则创建输出文件夹if not os.path.exists(output_path):os.makedirs(output_path)# 分割视频为照片print("beginning to extract frames from video !")for i in tqdm(range(total_frames)):ret, frame = cap.read()if ret:# 保存照片filename = f'./frame_%04d.png' % ifilepath = os.path.join(output_path, filename)cv2.imwrite(filepath, frame)else:break

2、图片指定区域批量去噪

可以参考我写的上篇博客，图象去噪。但是上篇博客只是对单个图片全图去噪，这次我们需要的是对一个文件夹里的图片指定区域进行批量去噪。

2.1、图象指定区域去噪

本次的任务是对图象下半部分进行中值滤波，故只需将原图裁剪一般得到下张图，对下半张图进行滤波后，与未滤波的上半张图像进行拼接，可使用纵向堆叠的np.vstack()实现图象上下拼接。

def mid_move_noise(noise_image): # 对图像下半部分进行中值去噪image = cv2.imread(noise_image)h, w = image.shape[:2]half_height = h // 2bottom_half = image[half_height:, :]bottom_half = cv2.medianBlur(bottom_half, 9)image = np.vstack((image[:half_height, :], bottom_half))return image

2.2、写一个for 循环遍历文件夹中的图片进行图像指定区域去噪，将去噪后的图片写入到指定文件夹下。

def from_fold_process_images(source_folder, result_folder):print("image process from folder: " + source_folder)for img_name in tqdm(os.listdir(source_folder)):img_path = os.path.join(source_folder, img_name)if img_name.endswith('.png'):# 构建输出图片的路径output_path = os.path.join(result_folder, img_name)# 应用中值滤波并保存img = mid_move_noise(img_path)cv2.imwrite(output_path, img)

3、去噪图片写入成视频

def merge_to_video(img_path, output_video_path):print("image merge to video")"""将图片合成视频参数：img_path 输入图片路径output_video_path 输出视频路径return 无"""# 传入输入路径中的所有图像image_files = [f for f in os.listdir(img_path) if f.endswith(".png")]image_files.sort()# 获取图像尺寸img = cv2.imread(os.path.join(img_path, image_files[0]))height, width, _ = img.shape# 创建用于写入视频的VideoWriter对象video_writer = cv2.VideoWriter(output_video_path, cv2.VideoWriter_fourcc(*"mp4v"), 30, (width, height))# 逐个读取图像并将其写入视频for image_file in tqdm(image_files):img = cv2.imread(os.path.join(img_path, image_file))video_writer.write(img)print("Video has been created successfully!")video_writer.release()

测试代码：

make_frame_img('./noise.mp4', './video_fps')
from_fold_process_images('./video_fps', './move_noise_fps')
merge_to_video('./move_noise_fps', './denoise.mp4')