未验证 提交 0afb4d5a 编写于 作者: W wjm 提交者: GitHub

fix export for FCOS (#8028)

上级 31500602
......@@ -262,23 +262,10 @@ class FCOSHead(nn.Layer):
bboxes_reg_list.append(bbox_reg)
centerness_list.append(centerness)
if targets is not None:
self.is_teacher = targets.get('ARSL_teacher', False)
if self.is_teacher:
return [cls_logits_list, bboxes_reg_list, centerness_list]
if targets is not None:
self.is_student = targets.get('ARSL_student', False)
if self.is_student:
return [cls_logits_list, bboxes_reg_list, centerness_list]
if targets is not None:
self.is_teacher = targets.get('is_teacher', False)
if self.is_teacher:
return [
locations_list, cls_logits_list, bboxes_reg_list,
centerness_list
]
return [cls_logits_list, bboxes_reg_list, centerness_list]
if self.training and targets is not None:
get_data = targets.get('get_data', False)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册