Skip to content

GitLab

Commit a5ae0d64 authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #487 from amirbar/master 

Fix issue #459
parents 7eb16d1f fa7ce18b
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
slim/nets/resnet_v1.py
View file @ a5ae0d64
...@@ -226,6 +226,7 @@ def resnet_v1_50(inputs, ...@@ -226,6 +226,7 @@ def resnet_v1_50(inputs,
return resnet_v1(inputs, blocks, num_classes, is_training, return resnet_v1(inputs, blocks, num_classes, is_training,
global_pool=global_pool, output_stride=output_stride, global_pool=global_pool, output_stride=output_stride,
include_root_block=True, reuse=reuse, scope=scope) include_root_block=True, reuse=reuse, scope=scope)
resnet_v1_50.default_image_size = resnet_v1.default_image_size
def resnet_v1_101(inputs, def resnet_v1_101(inputs,
...@@ -249,6 +250,7 @@ def resnet_v1_101(inputs, ...@@ -249,6 +250,7 @@ def resnet_v1_101(inputs,
return resnet_v1(inputs, blocks, num_classes, is_training, return resnet_v1(inputs, blocks, num_classes, is_training,
global_pool=global_pool, output_stride=output_stride, global_pool=global_pool, output_stride=output_stride,
include_root_block=True, reuse=reuse, scope=scope) include_root_block=True, reuse=reuse, scope=scope)
resnet_v1_101.default_image_size = resnet_v1.default_image_size
def resnet_v1_152(inputs, def resnet_v1_152(inputs,
...@@ -271,6 +273,7 @@ def resnet_v1_152(inputs, ...@@ -271,6 +273,7 @@ def resnet_v1_152(inputs,
return resnet_v1(inputs, blocks, num_classes, is_training, return resnet_v1(inputs, blocks, num_classes, is_training,
global_pool=global_pool, output_stride=output_stride, global_pool=global_pool, output_stride=output_stride,
include_root_block=True, reuse=reuse, scope=scope) include_root_block=True, reuse=reuse, scope=scope)
resnet_v1_152.default_image_size = resnet_v1.default_image_size
def resnet_v1_200(inputs, def resnet_v1_200(inputs,
...@@ -293,3 +296,4 @@ def resnet_v1_200(inputs, ...@@ -293,3 +296,4 @@ def resnet_v1_200(inputs,
return resnet_v1(inputs, blocks, num_classes, is_training, return resnet_v1(inputs, blocks, num_classes, is_training,
global_pool=global_pool, output_stride=output_stride, global_pool=global_pool, output_stride=output_stride,
include_root_block=True, reuse=reuse, scope=scope) include_root_block=True, reuse=reuse, scope=scope)
resnet_v1_200.default_image_size = resnet_v1.default_image_size
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment