示例#1
0
 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':
示例#4
0
 def load_test_data(self):
     mydata = DataProcess(self.img_rows, self.img_cols)
     imgs_test = mydata.load_test_data()
     return imgs_test