From 797f4fba027118bebd68a80ff58ec7d41fc590d6 Mon Sep 17 00:00:00 2001
From: riki <rgall@sertiscorp.com>
Date: Thu, 12 Sep 2019 09:50:32 +0700
Subject: [PATCH] Fix convert_crop Caffe

Add support for len(param.offset)==1


Removed axis from crop op args
---
 tools/python/transform/caffe_converter.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tools/python/transform/caffe_converter.py b/tools/python/transform/caffe_converter.py
index aa28cfb0..a185755b 100644
--- a/tools/python/transform/caffe_converter.py
+++ b/tools/python/transform/caffe_converter.py
@@ -553,13 +553,16 @@ class CaffeConverter(base_converter.ConverterInterface):
         param = caffe_op.layer.crop_param
         op.type = MaceOp.Crop.name
 
-        axis = param.axis
+        axis = 2
+        if param.HasField('axis'):
+            axis = param.axis
         axis = 4 + axis if axis < 0 else axis
+
         offset_value = -1 * np.ones(4, dtype=np.int32)
         offset_len = len(param.offset)
-        if offset_len == 1:
+        if offset_len <= 1:
             while axis < 4:
-                offset_value[axis] = param.offset[0]
+                offset_value[axis] = 0 if offset_len < 1 else param.offset[0]
                 axis += 1
         else:
             offset_value[axis:] = param.offset
-- 
GitLab