Skip to content

GitLab

Commit 0333144f authored by Neal Wu's avatar Neal Wu
Browse files

Set up resnet default_image_size code better

parent c1426f30
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 4 deletions
slim/nets/resnet_v2.py
View file @ 0333144f
...@@ -212,6 +212,7 @@ def resnet_v2(inputs, ...@@ -212,6 +212,7 @@ def resnet_v2(inputs,
if num_classes is not None: if num_classes is not None:
end_points['predictions'] = slim.softmax(net, scope='predictions') end_points['predictions'] = slim.softmax(net, scope='predictions')
return logits, end_points return logits, end_points
resnet_v2.default_image_size = 224
def resnet_v2_50(inputs, def resnet_v2_50(inputs,
...@@ -234,7 +235,8 @@ def resnet_v2_50(inputs, ...@@ -234,7 +235,8 @@ def resnet_v2_50(inputs,
return resnet_v2(inputs, blocks, num_classes, is_training=is_training, return resnet_v2(inputs, blocks, num_classes, is_training=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_v2_50.default_image_size = 224 resnet_v2_50.default_image_size = resnet_v2.default_image_size
def resnet_v2_101(inputs, def resnet_v2_101(inputs,
num_classes=None, num_classes=None,
...@@ -256,7 +258,7 @@ def resnet_v2_101(inputs, ...@@ -256,7 +258,7 @@ def resnet_v2_101(inputs,
return resnet_v2(inputs, blocks, num_classes, is_training=is_training, return resnet_v2(inputs, blocks, num_classes, is_training=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_v2_101.default_image_size = 224 resnet_v2_101.default_image_size = resnet_v2.default_image_size
def resnet_v2_152(inputs, def resnet_v2_152(inputs,
...@@ -279,7 +281,7 @@ def resnet_v2_152(inputs, ...@@ -279,7 +281,7 @@ def resnet_v2_152(inputs,
return resnet_v2(inputs, blocks, num_classes, is_training=is_training, return resnet_v2(inputs, blocks, num_classes, is_training=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_v2_152.default_image_size = 224 resnet_v2_152.default_image_size = resnet_v2.default_image_size
def resnet_v2_200(inputs, def resnet_v2_200(inputs,
...@@ -302,4 +304,4 @@ def resnet_v2_200(inputs, ...@@ -302,4 +304,4 @@ def resnet_v2_200(inputs,
return resnet_v2(inputs, blocks, num_classes, is_training=is_training, return resnet_v2(inputs, blocks, num_classes, is_training=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_v2_200.default_image_size = 224 resnet_v2_200.default_image_size = resnet_v2.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