diff --git a/configs/htc/htc_r50_fpn_1x.yml b/configs/htc/htc_r50_fpn_1x.yml index 9b2f6b8b7307c5dd504e3271fac2ade428a767de..348343ccf4dbc511cfe2a98e80d97dad33572743 100644 --- a/configs/htc/htc_r50_fpn_1x.yml +++ b/configs/htc/htc_r50_fpn_1x.yml @@ -212,3 +212,14 @@ TestReader: - !ResizeImage interp: 1 max_size: 1333 + target_size: 800 + use_cv2: true + - !Permute + channel_first: true + to_bgr: false + batch_transforms: + - !PadBatch + pad_to_stride: 32 + use_padded_im_info: false + batch_size: 1 + shuffle: false diff --git a/ppdet/modeling/architectures/htc.py b/ppdet/modeling/architectures/htc.py index 97f42ebdd6453587f56f19962caae9cdf3114b00..16c2909161265ceb0abbf0de44355867ccae37a2 100644 --- a/ppdet/modeling/architectures/htc.py +++ b/ppdet/modeling/architectures/htc.py @@ -467,5 +467,7 @@ class HybridTaskCascade(object): return self.build_multi_scale(feed_vars, mask_branch) return self.build(feed_vars, 'test') - def test(self, feed_vars): + def test(self, feed_vars, exclude_nms=False): + assert not exclude_nms, "exclude_nms for {} is not support currently".format( + self.__class__.__name__) return self.build(feed_vars, 'test')