소스 검색

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,