Args: point_clouds: (batch_size, num_point, 3) Returns: pointclouds_pred: (batch_size, num_point, 3)
(X_inputs, Y_inputs, is_training, bn_decay=None, weight_decay=None)
| 13 | |
| 14 | |
| 15 | def get_model(X_inputs, Y_inputs, is_training, bn_decay=None, weight_decay=None): |
| 16 | """ |
| 17 | Args: |
| 18 | point_clouds: (batch_size, num_point, 3) |
| 19 | Returns: |
| 20 | pointclouds_pred: (batch_size, num_point, 3) |
| 21 | """ |
| 22 | batch_size = X_inputs.get_shape()[0].value |
| 23 | num_point = X_inputs.get_shape()[1].value |
| 24 | nu.args.phase = is_training |
| 25 | print(is_training) |
| 26 | |
| 27 | X_feats = nu.create_pcn_encoder(X_inputs, name='X') |
| 28 | pred_X = nu.create_decoder(X_feats, name='X') |
| 29 | X2Y_feats, X2Y_code = nu.create_transferer_X2Y(X_feats, name='X2Y') |
| 30 | print(X2Y_feats) |
| 31 | print(X2Y_code) |
| 32 | X2Y2X_feats, _ = nu.create_transferer_Y2X(X2Y_feats, X2Y_code, name='Y2X') |
| 33 | pred_X2Y2X = nu.create_decoder(X2Y2X_feats, name='X') |
| 34 | complete_X = nu.create_decoder(X2Y_feats, name='Y') |
| 35 | |
| 36 | |
| 37 | Y_feats = nu.create_pcn_encoder(Y_inputs, name='Y') |
| 38 | pred_Y = nu.create_decoder(Y_feats, name='Y') |
| 39 | Y2X_feats, Y2X_code = nu.create_transferer_Y2X(Y_feats, None, name='Y2X') |
| 40 | Y2X2Y_feats, Y2X2Y_code = nu.create_transferer_X2Y(Y2X_feats, name='X2Y') |
| 41 | pred_Y2X2Y = nu.create_decoder(Y2X2Y_feats, name='Y') |
| 42 | incomplete_Y = nu.create_decoder(Y2X_feats, name='X') |
| 43 | |
| 44 | X2Y_logits = nu.create_discrminator(X2Y_feats, name='Y') |
| 45 | Y_logits = nu.create_discrminator(Y_feats, name='Y') |
| 46 | |
| 47 | Y2X_logits = nu.create_discrminator(Y2X_feats, name='X') |
| 48 | X_logits = nu.create_discrminator(X_feats, name='X') |
| 49 | |
| 50 | return pred_X, pred_Y, pred_Y2X2Y, pred_X2Y2X, X2Y_logits, Y_logits, Y2X_logits, X_logits,\ |
| 51 | X_feats, X2Y_feats, Y_feats, Y2X_feats, complete_X, incomplete_Y, Y2X2Y_feats, X2Y2X_feats, X2Y_code, Y2X_code, Y2X2Y_code |
| 52 | |
| 53 | def get_loss(pred_X, pred_Y, pred_Y2X2Y, pred_X2Y2X, X2Y_logits, Y_logits, Y2X_logits, X_logits, X_feats, X2Y_feats,\ |
| 54 | Y_feats, Y2X_feats, complete_X, incomplete_Y, gt_X, gt_Y, gt_GT, Y2X2Y_feats, X2Y2X_feats, X2Y_code, Y2X_code, Y2X2Y_code): |
nothing calls this directly
no outgoing calls
no test coverage detected