diff --git a/ppdet/data/data_feed.py b/ppdet/data/data_feed.py index c384b2cb3241cc5c012cedc02dfc9cbeab524bf6..d2f4edfbc60bf235aa570410d2782daaeae75a9e 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()],