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]
