SRGAN-tensorflow icon indicating copy to clipboard operation
SRGAN-tensorflow copied to clipboard

KeyError: 'vgg19_1/vgg_19/conv5/conv5_4'

Open youthM opened this issue 6 years ago • 7 comments

Traceback (most recent call last): File "main.py", line 231, in Net = SRGAN(data.inputs, data.targets, FLAGS) File "/home/zhou/SRGAN-tensorflow/lib/model.py", line 372, in SRGAN extracted_feature_gen = VGG19_slim(gen_output, FLAGS.perceptual_mode, reuse=False, scope=scope) File "/home/zhou/SRGAN-tensorflow/lib/model.py", line 341, in VGG19_slim output = output[target_layer] KeyError: 'vgg19_1/vgg_19/conv5/conv5_4'

youthM avatar Apr 10 '18 07:04 youthM

Can you tell me the other SRGAN_pre-trained , such as model _100000 @brade31919

youthM avatar Apr 10 '18 07:04 youthM

@youthM same issue in the third training step, have you resolved it?

ShiinaMitsuki avatar Jun 20 '18 09:06 ShiinaMitsuki

On line of 314 of model.py you have to change target_layer = scope + 'vgg_19/conv5/conv5_4' to target_layer = 'vgg_19/conv5/conv5_4'

mhurliman avatar Jun 20 '18 15:06 mhurliman

Or 316 similarly.

mhurliman avatar Jun 20 '18 15:06 mhurliman

@youthM same issue in the third training step, have you resolved it?

I haven't resolved it.

youthM avatar Oct 26 '18 13:10 youthM

On line of 314 of model.py you have to change target_layer = scope + 'vgg_19/conv5/conv5_4' to target_layer = 'vgg_19/conv5/conv5_4'

I tried but failed

youthM avatar Oct 26 '18 13:10 youthM

I tried but failed!!

lijain avatar Aug 07 '20 09:08 lijain