import numpy as np import os import torch import cv2 from net.dornet import Net from data.rgbdd_dataloader import * from PIL import Image import torchvision.transforms as transforms # RGBDD_Dataset data process # S1: load data lr_path = r"C:\Users\wuyuan\Downloads/model_out.png" lr_pil = Image.open(lr_path) import pdb;pdb.set_trace()