facade_dataset.py 文件源码

def __init__(self, dataDir='./datasets/facade/base', data_start = 1, data_end = 300):

        self.dataset = []
        self.shufflelist = []

        for i in range (data_start, data_end + 1):
            real_image = cv2.imread(dataDir+"/cmp_b%04d.jpg"%i)
            real_image = self.resize(real_image, ipl_alg = cv2.INTER_CUBIC)

            input_x_image = cv2.imread(dataDir+"/cmp_b%04d.png"%i)
            input_x_image = self.resize(input_x_image, ipl_alg = cv2.INTER_NEAREST)

            self.dataset.append((input_x_image, real_image))

            self.shufflelist = list(range(self.len()))