File size: 1,012 Bytes
db81e17 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | {
"model_type": "CycleGAN",
"checkpoint_file": "checkpoint.pt",
"checkpoint_keys": {
"model_state_dict": "model_state_dict"
},
"domains": {
"A": "apple",
"B": "orange"
},
"directions": {
"A_to_B": "apple_to_orange",
"B_to_A": "orange_to_apple"
},
"image": {
"image_size": 256,
"channels": 3
},
"preprocessing": {
"resize": null,
"center_crop": null,
"to_tensor": true
},
"normalization": {
"mean_A": [0.5921, 0.4201, 0.3659],
"std_A": [0.2339, 0.2732, 0.2590],
"mean_B": [0.6526, 0.4834, 0.2991],
"std_B": [0.2155, 0.2118, 0.2252]
},
"postprocessing": {
"denormalize_A": {
"mean": [0.5921, 0.4201, 0.3659],
"std": [0.2339, 0.2732, 0.2590]
},
"denormalize_B": {
"mean": [0.6526, 0.4834, 0.2991],
"std": [0.2155, 0.2118, 0.2252]
}
},
"model_params": {
"gen_d_base": 64,
"gen_n_down": 2,
"gen_n_blocks": 9,
"dis_d_in": 3,
"dis_d_base": 64,
"dis_n_down": 3
}
} |