图像处理：空白patch过滤（空白图判断）

WiSirius

已于 2024-02-04 14:52:51 修改

阅读量883

点赞数 13

分类专栏：图像处理文章标签：图像处理人工智能

于 2024-01-09 16:04:05 首次发布

本文链接：https://blog.csdn.net/WiSirius/article/details/135481707

版权

前言

在进行训练相关任务时，有时会对图像进行裁剪或取patch的操作，进行这种操作后，不可避免的会引入一些空白patch，或内容量很少的图像，这种图像通常会干扰模型的训练，因此最好过滤掉。本文提供了3种方法对图像进行过滤。

如果一张图像为空，或者信息很少，其像素的种类也很少，可以根据这一特点进行判断。通过将唯一像素计数与阈值比较，我们可以确定空图像。可以根据阈值来决定过滤的程度。

import numpy as np
import cv2
def is_empty(img):
   # Reading Image
   image = cv2.imread(img, 0)
   np.reshape(image, (-1,1))
   u, count_unique