amirbar 9 년 전
부모
커밋
780ecbb3c9
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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