amirbar hace 9 años
padre
commit
780ecbb3c9
Se han modificado 1 ficheros con 2 adiciones y 2 borrados
  1. 2 2
      slim/nets/resnet_v1.py

+ 2 - 2
slim/nets/resnet_v1.py

@@ -226,7 +226,7 @@ def resnet_v1_50(inputs,
   return resnet_v1(inputs, blocks, num_classes, is_training,
                    global_pool=global_pool, output_stride=output_stride,
                    include_root_block=True, reuse=reuse, scope=scope)
-
+resnet_v1_50.default_image_size = resnet_v1.default_image_size
 
 def resnet_v1_101(inputs,
                   num_classes=None,
@@ -293,4 +293,4 @@ def resnet_v1_200(inputs,
   return resnet_v1(inputs, blocks, num_classes, is_training,
                    global_pool=global_pool, output_stride=output_stride,
                    include_root_block=True, reuse=reuse, scope=scope)
-resnet_v1_200.default_image_size = resnet_v1.default_image_size
+resnet_v1_200.default_image_size = resnet_v1.default_image_size