""" 测试脚本 - DDD增强网络推理 只保存预测结果,不计算指 作者: Dxxx Dexxx 日期: 2025 """ import os import sys import argparse import torch import cv2 import numpy as np from tqdm import tqdm from pathlib import Path # 添加上级目录到路径 from data import DDDEnhancerDataset def function(): """ 保存预测结果 Args: pred: 预测结果张量 [1, H, W] 或 [H, W],值在[0, 1] save_path: 保存路径 original_size: 原始图像尺寸 (height, width),如果提供则调整到此尺寸 """ pass