initial commit

This commit is contained in:
2026-03-22 03:21:45 +02:00
commit 897fea9f4e
15431 changed files with 2548840 additions and 0 deletions

View File

@@ -0,0 +1,32 @@
"""
测试脚本 - 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