implement crop operator of TensorflowEmitter. Keras -> Tensorflow nasnet test passed.

This commit is contained in:
Kit 2018-02-07 13:43:48 +08:00
Родитель e04910c040
Коммит b415ffc034
2 изменённых файлов: 16 добавлений и 1 удалений

Просмотреть файл

@ -403,6 +403,21 @@ def KitModel(weight_file = None):
padding,
IR_node.name))
def emit_Crop(self, IR_node):
border = IR_node.get_attr('border')
assert len(border) == 4
output_shape = IR_node.get_attr('_output_shapes')[0]
output_shape = shape_to_list(output_shape)
self.add_body(1, "{:<15} = tf.image.crop_to_bounding_box({}, offset_height={}, offset_width={}, target_height={}, target_width={})".format(
IR_node.variable_name,
self.parent_variable_name(IR_node),
border[0],
border[2],
output_shape[1],
output_shape[2]))
def _layer_Conv(self):
self.add_body(0, """

Просмотреть файл

@ -200,7 +200,7 @@ class TestModels(CorrectnessTest):
'densenet' : [CntkEmit, TensorflowEmit, KerasEmit],
'xception' : [TensorflowEmit, KerasEmit],
'mobilenet' : [TensorflowEmit, KerasEmit],
'nasnet' : [KerasEmit],
'nasnet' : [TensorflowEmit, KerasEmit],
}
}