Closed YefeiGao closed 4 years ago
i modified three parts of the code and it works, that inclued:
with slim.arg_scope(resnet_v1.resnet_arg_scope(weight_decay=weight_decay)):
# logits, end_points = resnet_v1.resnet_v1_50(images, is_training=is_training, scope='resnet_v1_50')
logits, end_points = resnet_v1.resnet_v1_101(images, is_training=is_training, scope='resnet_v1_101')
if __name__ == '__main__':
input = tf.placeholder(tf.float32, shape=(None, 224, 224, 3), name='input')
with slim.arg_scope(resnet_arg_scope()) as sc:
# logits = resnet_v1_50(input)
logits = resnet_v1_101(input)
try:
# end_points['pool3'] = end_points['resnet_v1_50/block1']
# end_points['pool4'] = end_points['resnet_v1_50/block2']
end_points['pool3'] = end_points['resnet_v1_101/block1']
end_points['pool4'] = end_points['resnet_v1_101/block2']
except:
end_points['pool3'] = end_points['Detection/resnet_v1_50/block1']
end_points['pool4'] = end_points['Detection/resnet_v1_50/block2']
hope this help others :)
Hello @YefeiGao
You are right.
And you managed to solve it before I saw your email, hahah. :)
Thanks for ur work on this project, when i use resnet101 backbone based pre-trained model by replace the checkpoint path, it just didn't work, so what can i do to test resnet101 pre-trained model? Thx!