def test_build_feature_network(self):
config = hparams_config.get_efficientdet_config('efficientdet-d0')
with tf.Session(graph=tf.Graph()) as sess:
inputs = {
0: tf.ones([1, 512, 512, 3]),
1: tf.ones([1, 256, 256, 16]),
2: tf.ones([1, 128, 128, 24]),
3: tf.ones([1, 64, 64, 40]),
4: tf.ones([1, 32, 32, 112]),
5: tf.ones([1, 16, 16, 320])
}
tf.random.set_random_seed(SEED)
new_feats1 = efficientdet_arch_keras.build_feature_network(
inputs, config)
sess.run(tf.global_variables_initializer())
new_feats1 = sess.run(new_feats1)
with tf.Session(graph=tf.Graph()) as sess:
inputs = {
0: tf.ones([1, 512, 512, 3]),
1: tf.ones([1, 256, 256, 16]),
2: tf.ones([1, 128, 128, 24]),
3: tf.ones([1, 64, 64, 40]),
4: tf.ones([1, 32, 32, 112]),
5: tf.ones([1, 16, 16, 320])
}
tf.random.set_random_seed(SEED)
new_feats2 = legacy_arch.build_feature_network(inputs, config)
sess.run(tf.global_variables_initializer())
new_feats2 = sess.run(new_feats2)
for i in range(config.min_level, config.max_level + 1):
self.assertAllEqual(new_feats1[i], new_feats2[i])
def test_build_feature_network(self):
config = hparams_config.get_efficientdet_config('efficientdet-d0')
config.max_level = 5
with tf.Session(graph=tf.Graph()) as sess:
inputs = [
tf.ones([1, 64, 64, 40]), # level 3
tf.ones([1, 32, 32, 112]), # level 4
tf.ones([1, 16, 16, 320]), # level 5
]
tf.random.set_random_seed(SEED)
new_feats1 = efficientdet_keras.FPNCells(config)(inputs, True)
sess.run(tf.global_variables_initializer())
keras_feats = sess.run(new_feats1)
with tf.Session(graph=tf.Graph()) as sess:
inputs = {
0: tf.ones([1, 512, 512, 3]),
1: tf.ones([1, 256, 256, 16]),
2: tf.ones([1, 128, 128, 24]),
3: tf.ones([1, 64, 64, 40]),
4: tf.ones([1, 32, 32, 112]),
5: tf.ones([1, 16, 16, 320])
}
tf.random.set_random_seed(SEED)
new_feats2 = legacy_arch.build_feature_network(inputs, config)
sess.run(tf.global_variables_initializer())
legacy_feats = sess.run(new_feats2)
for i in range(config.min_level, config.max_level + 1):
self.assertAllClose(keras_feats[i - config.min_level], legacy_feats[i])
def efficientdet(features, model_name=None, config=None, **kwargs):
"""Build EfficientDet model.
Args:
features: input tensor.
model_name: String of the model (eg. efficientdet-d0)
config: Dict of parameters for the network
**kwargs: other parameters.
Returns:
A tuple (class_outputs, box_outputs) for predictions.
"""
if not config and not model_name:
raise ValueError('please specify either model name or config')
if not config:
config = hparams_config.get_efficientdet_config(model_name)
elif isinstance(config, dict):
config = hparams_config.Config(config) # wrap dict in Config object
if kwargs:
config.override(kwargs)
logging.info(config)
# build backbone features.
features = legacy_arch.build_backbone(features, config)
logging.info('backbone params/flops = {:.6f}M, {:.9f}B'.format(
*utils.num_params_flops()))
# build feature network.
fpn_feats = legacy_arch.build_feature_network(features, config)
logging.info('backbone+fpn params/flops = {:.6f}M, {:.9f}B'.format(
*utils.num_params_flops()))
# build class and box predictions.
class_box = BuildClassAndBoxOutputs(**config)
class_outputs, box_outputs = class_box.call(fpn_feats)
logging.info('backbone+fpn+box params/flops = {:.6f}M, {:.9f}B'.format(
*utils.num_params_flops()))
return class_outputs, box_outputs
def test_model_output(self):
inputs_shape = [1, 512, 512, 3]
config = hparams_config.get_efficientdet_config('efficientdet-d0')
with tf.Session(graph=tf.Graph()) as sess:
feats = tf.ones(inputs_shape)
tf.random.set_random_seed(SEED)
feats, _ = efficientdet_arch_keras.build_backbone(feats, config)
feats = efficientdet_arch_keras.build_feature_network(
feats, config)
feats = efficientdet_arch_keras.build_class_and_box_outputs(
feats, config)
sess.run(tf.global_variables_initializer())
class_output1, box_output1 = sess.run(feats)
with tf.Session(graph=tf.Graph()) as sess:
feats = tf.ones(inputs_shape)
tf.random.set_random_seed(SEED)
feats = legacy_arch.build_backbone(feats, config)
feats = legacy_arch.build_feature_network(feats, config)
feats = legacy_arch.build_class_and_box_outputs(feats, config)
sess.run(tf.global_variables_initializer())
class_output2, box_output2 = sess.run(feats)
for i in range(3, 8):
self.assertAllEqual(class_output1[i], class_output2[i])
self.assertAllEqual(box_output1[i], box_output2[i])
