(tensor)
| 46 | return num_params / 1e6 |
| 47 | |
| 48 | def tensor2numpy(tensor): |
| 49 | tensor = tensor * 0.5 + 0.5 |
| 50 | tensor = tensor.squeeze(0).permute(1, 2, 0).flip(2) |
| 51 | return np.clip(tensor.float().cpu().numpy(), 0, 1) * 255.0 |
| 52 | |
| 53 | |
| 54 | class MARCONetPlus(object): |