Commit 62c4fa1a authored by Jiangxin Dong's avatar Jiangxin Dong
Browse files

Upload New File

parent aa1580c6
import os
from data import vsrdata
class Benchmark(vsrdata.VSRData):
"""
Data generator for benchmark tasks
"""
def __init__(self, args, name='', train=False):
super(Benchmark, self).__init__(args, name=name, train=train)
def _set_filesystem(self, dir_data):
if self.args.template == "SVMAP" :
self.apath = os.path.join(dir_data)
if not self.args.test_only:
self.dir_image_blur = os.path.join(self.apath, "blurredImage")
self.dir_image_gt = os.path.join(self.apath, "GTImage")
self.dir_image_kernel = os.path.join(self.apath, "kernelImage")
print("validation image path :" , self.dir_image_blur)
else:
self.dir_image_blur = os.path.join(self.apath, "blurredImage")
self.dir_image_gt = os.path.join(self.apath, "blurredImage")
self.dir_image_kernel = os.path.join(self.apath, "kernelImage")
print("validation 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