From 5dce86c64d1483136c34226aa707ac9eddf5b27b Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Thu, 1 Mar 2018 11:38:40 +0800
Subject: [PATCH] Fix atrous convolution transform bug.

---
 python/tools/model.template          | 12 ++++++------
 python/tools/source_converter_lib.py | 11 ++++++-----
 python/tools/tf_converter_lib.py     |  1 +
 3 files changed, 13 insertions(+), 11 deletions(-)

diff --git a/python/tools/model.template b/python/tools/model.template
index aad28d08..f4826b7c 100644
--- a/python/tools/model.template
+++ b/python/tools/model.template
@@ -15,8 +15,8 @@
 namespace mace {
 namespace {{tag}} {
 
-void Create{{tensor.name}}(std::vector<mace::ConstTensor> &tensors,
-                           const unsigned char *model_data) {
+void CreateTensor{{tensor_info.id}}(std::vector<mace::ConstTensor> &tensors,
+                               const unsigned char *model_data) {
   MACE_LATENCY_LOGGER(2, "Create tensor {{ tensor.name }}");
   tensors.emplace_back(mace::ConstTensor(
       {{ tensor.name|tojson }}, model_data + {{ offset }},
@@ -189,8 +189,8 @@ namespace mace {
 namespace {{tag}} {
 
 {% for tensor in tensors %}
-extern void Create{{ tensor.name }}(std::vector<mace::ConstTensor> &tensors,
-                                    const unsigned char *model_data);
+extern void CreateTensor{{ tensor.id }}(std::vector<mace::ConstTensor> &tensors,
+                                        const unsigned char *model_data);
 {% endfor %}
 
 
@@ -269,9 +269,9 @@ void CreateTensors(std::vector<mace::ConstTensor> &tensors,
   MACE_LATENCY_LOGGER(1, "Create tensors");
   tensors.reserve({{ net.tensors|length }});
 
-  {% for tensor in net.tensors %}
+  {% for tensor in tensors %}
 
-  mace::{{tag}}::Create{{tensor.name}}(tensors, model_data);
+  mace::{{tag}}::CreateTensor{{tensor.id}}(tensors, model_data);
   {% endfor %}
 }
 
diff --git a/python/tools/source_converter_lib.py b/python/tools/source_converter_lib.py
index c7061b98..ad4f9607 100644
--- a/python/tools/source_converter_lib.py
+++ b/python/tools/source_converter_lib.py
@@ -73,7 +73,8 @@ def rename_tensor(net_def):
         op.output[i] = tensor_map[op.output[i]]
 
 class TensorInfo:
-  def __init__(self, t, runtime):
+  def __init__(self, id, t, runtime):
+    self.id = id
     self.name = t.name
     self.data_type = mace_pb2.DataType.Name(t.data_type)
     if t.data_type == mace_pb2.DT_FLOAT:
@@ -106,20 +107,20 @@ def convert_to_source(net_def, mode_pb_checksum, template, obfuscate, model_tag,
   j2_env = Environment(loader=FileSystemLoader(template_dir),
     trim_blocks=True)
   j2_env.filters['stringfy'] = stringfy
-  counter = 0
   output_dir = os.path.dirname(output) + '/'
   # generate tensor source files
   model_data = []
   offset = 0
+  counter = 0
   for t in net_def.tensors:
-    tensor_info = TensorInfo(t, runtime)
+    tensor_info = TensorInfo(counter, t, runtime)
     # align
     if tensor_info.data_type != 'DT_UINT8' and offset % 4 != 0:
       padding = 4 - offset % 4
       model_data.extend(bytearray([0] * padding))
       offset += padding
     source = j2_env.get_template(template_name).render(
-      tensor_info = TensorInfo(t, runtime),
+      tensor_info = tensor_info,
       tensor = t,
       tag = model_tag,
       mode = 0,
@@ -164,7 +165,7 @@ def convert_to_source(net_def, mode_pb_checksum, template, obfuscate, model_tag,
     counter += 1
 
   # generate model source files
-  tensors = [TensorInfo(t, runtime) for t in net_def.tensors]
+  tensors = [TensorInfo(i, net_def.tensors[i], runtime) for i in range(len(net_def.tensors))]
   source = j2_env.get_template(template_name).render(
     tensors = tensors,
     net = net_def,
diff --git a/python/tools/tf_converter_lib.py b/python/tools/tf_converter_lib.py
index fa27feb5..b8cc3849 100644
--- a/python/tools/tf_converter_lib.py
+++ b/python/tools/tf_converter_lib.py
@@ -641,6 +641,7 @@ class TFConverter(object):
     conv_op = self.tf_graph[op.name][0]
     op_def.name = conv_op.name
     op_def.type = conv_op.type
+    self.transpose_filter_tensor[get_input_tensor(conv_op, 1).name] = (0, 1, 3, 2)
     if self.device == 'gpu':
       op_def.input.extend([op.inputs[0].name])
       output_name = self.add_buffer_to_image(get_input_tensor(conv_op, 1).name, "CONV2D_FILTER")
-- 
GitLab