32 lines
582 B
Python
32 lines
582 B
Python
|
"""
|
|||
|
|
测试脚本 - 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
|