From a992fef40eeca9d7547253596eef3bdef39fad0b Mon Sep 17 00:00:00 2001 From: FlyingQianMM <245467267@qq.com> Date: Tue, 12 Nov 2019 14:39:56 +0800 Subject: [PATCH] Add ResizeImage in FasterRCNNTestFeed and MaskRCNNTestFeed (#25) --- ppdet/data/data_feed.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ppdet/data/data_feed.py b/ppdet/data/data_feed.py index c384b2cb3..d2f4edfbc 100644 --- a/ppdet/data/data_feed.py +++ b/ppdet/data/data_feed.py @@ -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()], -- GitLab