Commit 026859b0 authored by Jiangxin Dong's avatar Jiangxin Dong
Browse files

Upload New File

parent 62c4fa1a
import os
from data import vsrdata
# Data loader for blur images
class BLUR_IMAGE(vsrdata.VSRData):
def __init__(self, args, name='BLUR_IMAGE', train=True):
super(BLUR_IMAGE, self).__init__(args, name=name, train=train)
def _scan(self):
names_sharp, names_blur, names_kernel = super(BLUR_IMAGE, self)._scan()
return names_sharp, names_blur, names_kernel
def _set_filesystem(self, dir_data):
if self.args.template == "SVMAP" :
print("loading image...")
self.apath = os.path.join(dir_data)
self.dir_image_blur = os.path.join(self.apath, "InputBlurredImage")
self.dir_image_gt = os.path.join(self.apath, "InputTargetImage")
self.dir_image_kernel = os.path.join(self.apath, "psfMotionKernel")
print("Training image path:" , self.dir_image_blur)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment