(model, model_path, output_name='gen/gen')
| 120 | |
| 121 | |
| 122 | def sample(model, model_path, output_name='gen/gen'): |
| 123 | pred = PredictConfig( |
| 124 | session_init=SmartInit(model_path), |
| 125 | model=model, |
| 126 | input_names=['z'], |
| 127 | output_names=[output_name, 'z']) |
| 128 | pred = SimpleDatasetPredictor(pred, RandomZData((100, args.z_dim))) |
| 129 | for o in pred.get_result(): |
| 130 | o = o[0] + 1 |
| 131 | o = o * 128.0 |
| 132 | o = np.clip(o, 0, 255) |
| 133 | o = o[:, :, :, ::-1] |
| 134 | stack_patches(o, nr_row=10, nr_col=10, viz=True) |
| 135 | |
| 136 | |
| 137 | def get_args(default_batch=128, default_z_dim=100): |
no test coverage detected