# show_pkl.py import pickle import sys import torch sys.path.append('/home/ly/StyleCLIP-main/models/stylegan3/torch_utils') # # path = '/home/ly/StyleCLIP-main/models/stylegan3/torch_utils/stylegan3-r-afhqv2-512x512.pkl' # path='/root/……/aus_openface.pkl' pkl文件所在路径 # # f = open(path, 'rb') # data = pickle.load(f) # # print(data) # print(len(data)) # print(data.shape) with open('/home/ly/StyleCLIP-main/models/stylegan3/torch_utils/stylegan3-r-afhqv2-512x512.pkl', 'rb') as f: G = pickle.load(f)['G_ema'].cuda() # torch.nn.Module z = torch.randn([1, G.z_dim]).cuda() # latent codes c = None # class labels (not used in this example) img = G(z, c) # NCHW, float32, dynamic range [-1, +1], no truncation print(G) #输出 # Generator( # (synthesis): SynthesisNetwork( # w_dim=512, num_ws=16, # img_resolution=512, img_channels=3, # num_layers=14, num_critical=2, # margin_size=10, num_fp16_res=4 # (input): SynthesisInput( # w_dim=512, channels=1024, size=[36, 36], # sampling_rate=16, bandwidth=2 # (affine): FullyConnectedLayer(in_features=512, out_features=4, activation=linear) # ) # (L0_36_1024): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=False, # in_sampling_rate=16, out_sampling_rate=16, # in_cutoff=2, out_cutoff=2, # in_half_width=6, out_half_width=6, # in_size=[36, 36], out_size=[36, 36], # in_channels=1024, out_channels=1024 # (affine): FullyConnectedLayer(in_features=512, out_features=1024, activation=linear) # ) # (L1_36_1024): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=False, # in_sampling_rate=16, out_sampling_rate=16, # in_cutoff=2, out_cutoff=2.99661, # in_half_width=6, out_half_width=5.00339, # in_size=[36, 36], out_size=[36, 36], # in_channels=1024, out_channels=1024 # (affine): FullyConnectedLayer(in_features=512, out_features=1024, activation=linear) # ) # (L2_52_1024): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=False, # in_sampling_rate=16, out_sampling_rate=32, # in_cutoff=2.99661, out_cutoff=4.48985, # in_half_width=5.00339, out_half_width=11.5102, # in_size=[36, 36], out_size=[52, 52], # in_channels=1024, out_channels=1024 # (affine): FullyConnectedLayer(in_features=512, out_features=1024, activation=linear) # ) # (L3_52_1024): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=False, # in_sampling_rate=32, out_sampling_rate=32, # in_cutoff=4.48985, out_cutoff=6.72717, # in_half_width=11.5102, out_half_width=9.27283, # in_size=[52, 52], out_size=[52, 52], # in_channels=1024, out_channels=1024 # (affine): FullyConnectedLayer(in_features=512, out_features=1024, activation=linear) # ) # (L4_84_1024): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=True, # in_sampling_rate=32, out_sampling_rate=64, # in_cutoff=6.72717, out_cutoff=10.0794, # in_half_width=9.27283, out_half_width=21.9206, # in_size=[52, 52], out_size=[84, 84], # in_channels=1024, out_channels=1024 # (affine): FullyConnectedLayer(in_features=512, out_features=1024, activation=linear) # ) # (L5_84_1024): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=True, # in_sampling_rate=64, out_sampling_rate=64, # in_cutoff=10.0794, out_cutoff=15.102, # in_half_width=21.9206, out_half_width=16.898, # in_size=[84, 84], out_size=[84, 84], # in_channels=1024, out_channels=1024 # (affine): FullyConnectedLayer(in_features=512, out_features=1024, activation=linear) # ) # (L6_148_1024): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=True, # in_sampling_rate=64, out_sampling_rate=128, # in_cutoff=15.102, out_cutoff=22.6274, # in_half_width=16.898, out_half_width=41.3726, # in_size=[84, 84], out_size=[148, 148], # in_channels=1024, out_channels=1024 # (affine): FullyConnectedLayer(in_features=512, out_features=1024, activation=linear) # ) # (L7_148_967): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=True, # in_sampling_rate=128, out_sampling_rate=128, # in_cutoff=22.6274, out_cutoff=33.9028, # in_half_width=41.3726, out_half_width=30.0972, # in_size=[148, 148], out_size=[148, 148], # in_channels=1024, out_channels=967 # (affine): FullyConnectedLayer(in_features=512, out_features=1024, activation=linear) # ) # (L8_276_645): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=True, # in_sampling_rate=128, out_sampling_rate=256, # in_cutoff=33.9028, out_cutoff=50.7968, # in_half_width=30.0972, out_half_width=77.2032, # in_size=[148, 148], out_size=[276, 276], # in_channels=967, out_channels=645 # (affine): FullyConnectedLayer(in_features=512, out_features=967, activation=linear) # ) # (L9_276_431): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=True, # in_sampling_rate=256, out_sampling_rate=256, # in_cutoff=50.7968, out_cutoff=76.1093, # in_half_width=77.2032, out_half_width=51.8907, # in_size=[276, 276], out_size=[276, 276], # in_channels=645, out_channels=431 # (affine): FullyConnectedLayer(in_features=512, out_features=645, activation=linear) # ) # (L10_532_287): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=True, # in_sampling_rate=256, out_sampling_rate=512, # in_cutoff=76.1093, out_cutoff=114.035, # in_half_width=51.8907, out_half_width=141.965, # in_size=[276, 276], out_size=[532, 532], # in_channels=431, out_channels=287 # (affine): FullyConnectedLayer(in_features=512, out_features=431, activation=linear) # ) # (L11_532_192): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=False, use_fp16=True, # in_sampling_rate=512, out_sampling_rate=512, # in_cutoff=114.035, out_cutoff=170.86, # in_half_width=141.965, out_half_width=85.1405, # in_size=[532, 532], out_size=[532, 532], # in_channels=287, out_channels=192 # (affine): FullyConnectedLayer(in_features=512, out_features=287, activation=linear) # ) # (L12_532_128): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=True, use_fp16=True, # in_sampling_rate=512, out_sampling_rate=512, # in_cutoff=170.86, out_cutoff=256, # in_half_width=85.1405, out_half_width=59.173, # in_size=[532, 532], out_size=[532, 532], # in_channels=192, out_channels=128 # (affine): FullyConnectedLayer(in_features=512, out_features=192, activation=linear) # ) # (L13_512_128): SynthesisLayer( # w_dim=512, is_torgb=False, # is_critically_sampled=True, use_fp16=True, # in_sampling_rate=512, out_sampling_rate=512, # in_cutoff=256, out_cutoff=256, # in_half_width=59.173, out_half_width=59.173, # in_size=[532, 532], out_size=[512, 512], # in_channels=128, out_channels=128 # (affine): FullyConnectedLayer(in_features=512, out_features=128, activation=linear) # ) # (L14_512_3): SynthesisLayer( # w_dim=512, is_torgb=True, # is_critically_sampled=True, use_fp16=True, # in_sampling_rate=512, out_sampling_rate=512, # in_cutoff=256, out_cutoff=256, # in_half_width=59.173, out_half_width=59.173, # in_size=[512, 512], out_size=[512, 512], # in_channels=128, out_channels=3 # (affine): FullyConnectedLayer(in_features=512, out_features=128, activation=linear) # ) # ) # (mapping): MappingNetwork( # z_dim=512, c_dim=0, w_dim=512, num_ws=16 # (fc0): FullyConnectedLayer(in_features=512, out_features=512, activation=lrelu) # (fc1): FullyConnectedLayer(in_features=512, out_features=512, activation=lrelu) # ) # )