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