소스 검색

use qfl cls loss

yjh0410 2 년 전
부모
커밋
b4c63ade7d
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      config/model_config/rtcdet_config.py

+ 1 - 1
config/model_config/rtcdet_config.py

@@ -217,7 +217,7 @@ rtcdet_cfg = {
         'matcher_hpy': {'center_sampling_radius': 2.5,
                         'topk_candidate': 10},
         # ---------------- Loss config ----------------
-        'cls_loss': 'qfl',
+        'cls_loss': 'bce',
         'loss_cls_weight': 1.0,
         'loss_dfl_weight': 1.0,
         'loss_box_weight': 5.0,