未验证 提交 936ec224 编写于 作者: F Feng Ni 提交者: GitHub

add Resize export default interp (#6632)

上级 2607dbca
...@@ -92,6 +92,7 @@ def _parse_reader(reader_cfg, dataset_cfg, metric, arch, image_shape): ...@@ -92,6 +92,7 @@ def _parse_reader(reader_cfg, dataset_cfg, metric, arch, image_shape):
if key == 'Resize': if key == 'Resize':
if int(image_shape[1]) != -1: if int(image_shape[1]) != -1:
value['target_size'] = image_shape[1:] value['target_size'] = image_shape[1:]
value['interp'] = value.get('interp', 1) # cv2.INTER_LINEAR
if fuse_normalize and key == 'NormalizeImage': if fuse_normalize and key == 'NormalizeImage':
continue continue
p.update(value) p.update(value)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册