未验证 提交 52ecf506 编写于 作者: W wangguanzhong 提交者: GitHub

fix clip_bbox (#620)

上级 2b793cd8
...@@ -384,7 +384,7 @@ class YOLOv4Head(YOLOv3Head): ...@@ -384,7 +384,7 @@ class YOLOv4Head(YOLOv3Head):
yolo_loss=yolo_loss, yolo_loss=yolo_loss,
iou_aware=iou_aware, iou_aware=iou_aware,
iou_aware_factor=iou_aware_factor, iou_aware_factor=iou_aware_factor,
clip_box=clip_bbox) clip_bbox=clip_bbox)
self.spp_stage = spp_stage self.spp_stage = spp_stage
def _upsample(self, input, scale=2, name=None): def _upsample(self, input, scale=2, name=None):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册