svt_data_loader.py 文件源码

def nextBatch(self, batches, dataset='train'):
        imgH = 300
        imgW = 300
        if dataset == 'train':
            datalist = self.trainList
        if dataset == 'test':
            datalist = self.testList
        randomIndex = random.sample(range(len(datalist)), batches)
        images = []
        for index in randomIndex:
            fileName = './svt1/' + datalist[index][0]
            img = cv2.imread(fileName, cv2.IMREAD_COLOR)
            resized = cv2.resize(img, (imgW, imgH))
            resized = np.multiply(resized, 1.0/255.0)
            images.append(resized)
            anns.append(datalist[x][1])
        images = np.asarray(images)
        return (images, anns)