未验证 提交 d3bd5103 编写于 作者: 文幕地方's avatar 文幕地方 提交者: GitHub

fix bugs in DetResizeForTest

上级 3a0090dc
...@@ -119,10 +119,10 @@ class DetResizeForTest(object): ...@@ -119,10 +119,10 @@ class DetResizeForTest(object):
if 'image_shape' in kwargs: if 'image_shape' in kwargs:
self.image_shape = kwargs['image_shape'] self.image_shape = kwargs['image_shape']
self.resize_type = 1 self.resize_type = 1
if 'limit_side_len' in kwargs: elif 'limit_side_len' in kwargs:
self.limit_side_len = kwargs['limit_side_len'] self.limit_side_len = kwargs['limit_side_len']
self.limit_type = kwargs.get('limit_type', 'min') self.limit_type = kwargs.get('limit_type', 'min')
if 'resize_long' in kwargs: elif 'resize_long' in kwargs:
self.resize_type = 2 self.resize_type = 2
self.resize_long = kwargs.get('resize_long', 960) self.resize_long = kwargs.get('resize_long', 960)
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册