diff --git a/damo/base_models/core/end2end.py b/damo/base_models/core/end2end.py index 48a70a6..a5366c9 100644 --- a/damo/base_models/core/end2end.py +++ b/damo/base_models/core/end2end.py @@ -39,7 +39,7 @@ def forward( boxes, scores, background_class=-1, - box_coding=1, + box_coding=0, iou_threshold=0.45, max_output_boxes=100, plugin_version='1', @@ -63,7 +63,7 @@ def symbolic(g, boxes, scores, background_class=-1, - box_coding=1, + box_coding=0, iou_threshold=0.45, max_output_boxes=100, plugin_version='1', @@ -277,7 +277,7 @@ def __init__(self, super().__init__() self.device = device if device else torch.device('cpu') self.background_class = -1, - self.box_coding = 1, + self.box_coding = 0, self.iou_threshold = iou_thres self.max_obj = max_obj self.plugin_version = '1'