Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/d2go.git" did not exist on "7d35bae7a5ec574aa9a9c1f6d534a4082ca6ac4c"
Commit 0333144f authored by Neal Wu's avatar Neal Wu
Browse files

Set up resnet default_image_size code better

parent c1426f30
Hide 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,
if num_classes is not None:
end_points['predictions'] = slim.softmax(net, scope='predictions')
return logits, end_points
resnet_v2.default_image_size = 224
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,
global_pool=global_pool, output_stride=output_stride,
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,
num_classes=None,
......@@ -256,7 +258,7 @@ def resnet_v2_101(inputs,
return resnet_v2(inputs, blocks, num_classes, is_training=is_training,
global_pool=global_pool, output_stride=output_stride,
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,
......@@ -279,7 +281,7 @@ def resnet_v2_152(inputs,
return resnet_v2(inputs, blocks, num_classes, is_training=is_training,
global_pool=global_pool, output_stride=output_stride,
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,
......@@ -302,4 +304,4 @@ def resnet_v2_200(inputs,
return resnet_v2(inputs, blocks, num_classes, is_training=is_training,
global_pool=global_pool, output_stride=output_stride,
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