提交 8e239684 编写于 作者: D dengkaipeng

add comment

上级 b735e353
...@@ -272,6 +272,7 @@ class YoloLoss(fluid.dygraph.Layer): ...@@ -272,6 +272,7 @@ class YoloLoss(fluid.dygraph.Layer):
downsample = 32 downsample = 32
losses = [] losses = []
# YOLOv3 output fields is different between 'train' and 'eval' mode
if len(inputs) == 6: if len(inputs) == 6:
output1, output2, output3, gt_box, gt_label, gt_score = inputs output1, output2, output3, gt_box, gt_label, gt_score = inputs
elif len(inputs) == 8: elif len(inputs) == 8:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册