def getBatchData(self): _src_tr_img_batch, _src_tr_lab_batch = DA_init.next_batch(self.source_training_data[0], self.source_training_data[1], self.bs) _tar_tr_img_batch = DA_init.next_batch_unpaired(self.target_training_data, self.bs) feed_dict = {self.x_source: _src_tr_img_batch, self.y_source: _src_tr_lab_batch, self.x_target: _tar_tr_img_batch, self.is_training: True, self.keep_rate: 0.5} feed_dict_eval = {self.x_source: _src_tr_img_batch, self.y_source: _src_tr_lab_batch, self.x_target: _tar_tr_img_batch, self.is_training: False, self.keep_rate: 0.5} return feed_dict, feed_dict_eval
def getBatchData(self): _src_tr_img_batch, _src_tr_lab_batch = DA_init.next_batch( self.source_training_data[0], self.source_training_data[1], self.bs, data_aug=False) _tar_tr_img_batch = DA_init.next_batch_unpaired( self.target_training_data, self.bs, data_aug=False) feed_dict = { self.X: _src_tr_img_batch, self.Y: _tar_tr_img_batch, self.is_training: True } feed_dict_eval = { self.X: _src_tr_img_batch, self.Y: _tar_tr_img_batch, self.is_training: False } return feed_dict, feed_dict_eval