From b70501ab662d43eac927bf54a1497c0643762171 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Tue, 24 Mar 2020 10:11:46 +0800
Subject: [PATCH] fix msdef (#385)

fix multi-scale def
---
 ppdet/modeling/architectures/input_helper.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/ppdet/modeling/architectures/input_helper.py b/ppdet/modeling/architectures/input_helper.py
index 691f069ed..a6c961c89 100644
--- a/ppdet/modeling/architectures/input_helper.py
+++ b/ppdet/modeling/architectures/input_helper.py
@@ -20,11 +20,18 @@ def multiscale_def(image_shape, num_scale, use_flip=True):
     if use_flip:
         num_scale //= 2
         base_name_list.append('image_flip')
+        multiscale_def['image_flip'] = {
+            'shape': [None] + image_shape,
+            'dtype': 'float32',
+            'lod_level': 0
+        }
         multiscale_def['im_info_image_flip'] = {
             'shape': [None, 3],
             'dtype': 'float32',
             'lod_level': 0
         }
+        ms_def_names.append('image_flip')
+        ms_def_names.append('im_info_image_flip')
     for base_name in base_name_list:
         for i in range(0, num_scale - 1):
             name = base_name + '_scale_' + str(i)
-- 
GitLab