|  | 
 
 发表于 2024-11-27 16:24:21
|
显示全部楼层 
| 本帖最后由 wtxx8888 于 2024-11-27 16:26 编辑 
 if 'df' in self.archi_type:
 gpu_dst_code     = self.inter(self.encoder(warped_dst))
 gpu_pred_src_dst, gpu_pred_src_dstm = self.decoder_src(gpu_dst_code)
 _, gpu_pred_dst_dstm = self.decoder_dst(gpu_dst_code)
 
 elif 'liae' in self.archi_type:
 gpu_dst_code = self.encoder (warped_dst)
 gpu_dst_inter_B_code = self.inter_B (gpu_dst_code)
 gpu_dst_inter_AB_code = self.inter_AB (gpu_dst_code)
 gpu_dst_code = tf.concat([gpu_dst_inter_B_code,gpu_dst_inter_AB_code], nn.conv2d_ch_axis)
 gpu_src_dst_code = tf.concat([gpu_dst_inter_AB_code,gpu_dst_inter_AB_code], nn.conv2d_ch_axis)
 
 gpu_pred_src_dst, gpu_pred_src_dstm = self.decoder(gpu_src_dst_code)
 _, gpu_pred_dst_dstm = self.decoder(gpu_dst_code)
 
 蓝色的_是占位符,意思是第一组的数据不要,不赋值变量。只要第二组的数据。
 不光把数据合一了,还丢弃了蓝色占位符的那部分数据,神也变不回去NPY。
  | 
 |