浏览代码

add picodet

will-jl944 4 年之前
父节点
当前提交
7fd6c0ff28
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/cv/models/detector.py

+ 1 - 1
paddlex/cv/models/detector.py

@@ -686,7 +686,7 @@ class PicoDet(BaseDetector):
             loss_dfl = ppdet.modeling.DistributionFocalLoss(loss_weight=.25)
             loss_bbox = ppdet.modeling.GIoULoss(loss_weight=2.0)
             assigner = ppdet.modeling.SimOTAAssigner(
-                candidate_topk=10, iou_weight=6)
+                candidate_topk=10, iou_weight=6, num_classes=num_classes)
             nms = ppdet.modeling.MultiClassNMS(
                 nms_top_k=nms_top_k,
                 keep_top_k=nms_keep_top_k,