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

Add ResizeImage in FasterRCNNTestFeed and MaskRCNNTestFeed (#25)

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