From 5fe08d1fd36b0f4f18b4ef3ae099bde32422c272 Mon Sep 17 00:00:00 2001
From: wuchenghui <wuchenghui@xiaomi.com>
Date: Wed, 2 May 2018 16:01:22 +0800
Subject: [PATCH] fix tf converter bug

---
 mace/python/tools/tf_converter_lib.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mace/python/tools/tf_converter_lib.py b/mace/python/tools/tf_converter_lib.py
index d562228f..08107b55 100644
--- a/mace/python/tools/tf_converter_lib.py
+++ b/mace/python/tools/tf_converter_lib.py
@@ -406,7 +406,7 @@ class TFConverter(object):
         input_channels = weight_tensor_value.shape[2]
         # HWIO -> OIHW
         weight_tensor_value = weight_tensor_value.transpose(3, 2, 0, 1)
-        if input_shape[2] > 16 and input_shape[3] > 16:
+        if input_shape[1] > 16 and input_shape[2] > 16:
             G = np.array([
                 [1.0, 0.0, 0.0],
                 [-2.0 / 9, -2.0 / 9, -2.0 / 9],
-- 
GitLab