def load_train_data(self,mode = 'my_data'):
mydata = DataProcess(self.img_rows, self.img_cols)
if mode == 'my_data':
imgs_train, imgs_mask_train = mydata.load_my_train_data()
elif mode == 'pretrain':
imgs_train, imgs_mask_train = mydata.load_train_data()
else:
imgs_train, imgs_mask_train = mydata.load_small_train_data()
imgs_mask_train = to_categorical(imgs_mask_train, num_classes=2)
return imgs_train, imgs_mask_train
def load_test_data(self):
mydata = DataProcess(self.img_rows, self.img_cols)
def load_train_data(self,mode = 'my_data'):
mydata = DataProcess(self.img_rows, self.img_cols)
if mode == 'my_data':
def load_test_data(self):
mydata = DataProcess(self.img_rows, self.img_cols)
imgs_test = mydata.load_test_data()
return imgs_test