如何在Python中按验证规则跳过文件并继续遍历图像目录的循环？

我来帮你完善这个predict_data()函数，添加验证逻辑来跳过不符合要求的文件，同时让循环继续执行。咱们可以分几步来实现：

第一步：明确验证规则 #

首先得确定哪些文件需要跳过，常见的实用验证项包括：

• 文件扩展名是否为常见图像格式（比如.jpg, .png, .bmp, .jpeg）
• 图像是否能被成功读取（避免损坏或无效的文件）
• 可选：原始图像尺寸是否满足最小要求（防止resize后失真过于严重）

第二步：修改代码加入验证逻辑 #

下面是调整后的完整代码，每一步验证都加了注释说明：

import os
import cv2
from tqdm import tqdm

# 假设这些是你已经定义好的常量
PREDICT_DIR = "./your_predict_dir"
IMG_W_SIZE = 224
IMG_H_SIZE = 224

def predict_data():
    predict_data = []  # 修正了你原代码里的语法错误，初始化空列表
    # 定义允许的图像扩展名（统一转小写避免大小写问题）
    allowed_extensions = ('.jpg', '.jpeg', '.png', '.bmp', '.tif')
    
    for img in tqdm(os.listdir(PREDICT_DIR)):
        # 验证规则1：检查文件是否为合法图像格式
        if not img.lower().endswith(allowed_extensions):
            print(f"跳过非图像文件: {img}")
            continue
        
        path = os.path.join(PREDICT_DIR, img)
        
        # 验证规则2：检查图像是否能被成功读取
        im_gray = cv2.imread(path, cv2.IMREAD_GRAYSCALE)
        if im_gray is None:
            print(f"跳过损坏/无法读取的图像: {img}")
            continue
        
        # 可选验证规则3：检查原始图像尺寸是否达标
        if im_gray.shape[0] < IMG_H_SIZE or im_gray.shape[1] < IMG_W_SIZE:
            print(f"跳过尺寸过小的图像: {img}")
            continue
        
        # 执行你原本的图像处理计算
        im_gray_size = cv2.resize(im_gray, (IMG_W_SIZE, IMG_H_SIZE))
        (thresh, im_bw) = cv2.threshold(im_gray_size, 128, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)
        fm = cv2.Laplacian(im_gray_size, cv2.CV_64F).var()
        # 继续你的后续逻辑，比如将结果存入predict_data
        predict_data.append({"filename": img, "sharpness": fm})  # 示例：保存计算结果
    
    return predict_data

关键细节说明： #

• 用continue语句跳过不符合规则的文件，循环会直接进入下一次迭代，不会执行后续的图像处理代码
• 每个验证步骤都加了打印提示，方便你调试时快速定位被跳过的文件及原因
• 统一将文件名转小写后检查扩展名，避免因.JPG、.Png这类大小写差异导致的误判

你可以根据实际需求调整验证规则，比如添加图像通道数检查、特定像素值范围验证等。

内容的提问来源于stack exchange，提问作者Dinusha Thilakarathna