def test_returns_correct_shapes_with_conv_defs(self): image_height = 299 image_width = 299 depth_multiplier = 1.0 conv_def_block_12 = model_utils.ConvDefs(conv_name='conv_pw_12', filters=512) conv_def_block_13 = model_utils.ConvDefs(conv_name='conv_pw_13', filters=256) conv_defs = [conv_def_block_12, conv_def_block_13] expected_feature_map_shape = ( test_utils.moblenet_v1_expected_feature_map_shape_with_conv_defs) self._check_returns_correct_shape(image_height, image_width, depth_multiplier, expected_feature_map_shape, conv_defs=conv_defs)
def test_returns_correct_shapes_with_conv_defs( self): image_height = 299 image_width = 299 depth_multiplier = 1.0 conv_1 = model_utils.ConvDefs( conv_name='Conv_1', filters=256) conv_defs = [conv_1] expected_feature_map_shape = ( test_utils.moblenet_v2_expected_feature_map_shape_with_conv_defs) self._check_returns_correct_shape( 2, image_height, image_width, depth_multiplier, expected_feature_map_shape, conv_defs=conv_defs)
def _create_modified_mobilenet_config(): last_conv = model_utils.ConvDefs(conv_name='Conv_1', filters=256) return [last_conv]
def _create_modified_mobilenet_config(): conv_def_block_12 = model_utils.ConvDefs(conv_name='conv_pw_12', filters=512) conv_def_block_13 = model_utils.ConvDefs(conv_name='conv_pw_13', filters=256) return [conv_def_block_12, conv_def_block_13]