From 22c6e80aab3c5512fd3a461bbba405254a63d353 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Thu, 6 May 2021 13:34:10 +0800 Subject: [PATCH] support voc format in fcos & ttfnet (#2875) --- ppdet/data/transform/batch_operators.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/ppdet/data/transform/batch_operators.py b/ppdet/data/transform/batch_operators.py index 8d74ce0e8..936f442eb 100644 --- a/ppdet/data/transform/batch_operators.py +++ b/ppdet/data/transform/batch_operators.py @@ -452,9 +452,10 @@ class Gt2FCOSTarget(BaseOperator): sample['centerness{}'.format(lvl)] = np.reshape( ctn_targets_by_level[lvl], newshape=[grid_h, grid_w, 1]) - sample.pop('is_crowd') - sample.pop('gt_class') - sample.pop('gt_bbox') + sample.pop('is_crowd', None) + sample.pop('difficult', None) + sample.pop('gt_class', None) + sample.pop('gt_bbox', None) return samples @@ -531,11 +532,11 @@ class Gt2TTFTarget(BaseOperator): sample['ttf_heatmap'] = heatmap sample['ttf_box_target'] = box_target sample['ttf_reg_weight'] = reg_weight - sample.pop('is_crowd') - sample.pop('gt_class') - sample.pop('gt_bbox') - if 'gt_score' in sample: - sample.pop('gt_score') + sample.pop('is_crowd', None) + sample.pop('difficult', None) + sample.pop('gt_class', None) + sample.pop('gt_bbox', None) + sample.pop('gt_score', None) return samples def draw_truncate_gaussian(self, heatmap, center, h_radius, w_radius): -- GitLab