提交 a992fef4 编写于 作者: F FlyingQianMM 提交者: qingqing01

Add ResizeImage in FasterRCNNTestFeed and MaskRCNNTestFeed (#25)

上级 f137929f
...@@ -558,6 +558,7 @@ class FasterRCNNTestFeed(DataFeed): ...@@ -558,6 +558,7 @@ class FasterRCNNTestFeed(DataFeed):
std=[0.229, 0.224, 0.225], std=[0.229, 0.224, 0.225],
is_scale=True, is_scale=True,
is_channel_first=False), is_channel_first=False),
ResizeImage(target_size=800, max_size=1333, interp=1),
Permute(to_bgr=False) Permute(to_bgr=False)
], ],
batch_transforms=[PadBatch()], batch_transforms=[PadBatch()],
...@@ -704,6 +705,7 @@ class MaskRCNNTestFeed(DataFeed): ...@@ -704,6 +705,7 @@ class MaskRCNNTestFeed(DataFeed):
std=[0.229, 0.224, 0.225], std=[0.229, 0.224, 0.225],
is_scale=True, is_scale=True,
is_channel_first=False), is_channel_first=False),
ResizeImage(target_size=800, max_size=1333, interp=1),
Permute(to_bgr=False, channel_first=True) Permute(to_bgr=False, channel_first=True)
], ],
batch_transforms=[PadBatch()], batch_transforms=[PadBatch()],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册