tancik / StegaStamp

Invisible Hyperlinks in Physical Photographs
http://www.matthewtancik.com/stegastamp
MIT License
675 stars 191 forks source link

how to reducer the params for fc layer in decoder? #25

Open zimenglan-sysu-512 opened 3 years ago

zimenglan-sysu-512 commented 3 years ago

hi @tancik since the fc layer in decoder is too larger (1313 128 * 500 = 10,816,000), so i want use a conv(dim=1024) to increase the dimension and then use tf.reduce_mean(x, axis=[1, 2]), but in the first 1w iterations, the bit_acc always gets around 0.5, and the str_acc gets zero. so how to reduce the params for the fc layer, at the same time, it can get good results?

ggohem commented 3 years ago

hi @tancik since the fc layer in decoder is too larger (1313 128 * 500 = 10,816,000), so i want use a conv(dim=1024) to increase the dimension and then use tf.reduce_mean(x, axis=[1, 2]), but in the first 1w iterations, the bit_acc always gets around 0.5, and the str_acc gets zero. so how to reduce the params for the fc layer, at the same time, it can get good results?

Hey, is this problem solved? @zimenglan-sysu-512