DORNet / test_img.py
RaynWu2002's picture
Upload 49 files
68c537d verified
raw
history blame contribute delete
357 Bytes
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()