Update defaults.py

add default loss function
facebookresearch · Dec 23, 2024 · 0fd37da · 0fd37da
1 parent eb18286
commit 0fd37da
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/detectron2/config/defaults.py b/detectron2/config/defaults.py
@@ -654,3 +654,8 @@
 # Do not commit any configs into it.
 _C.GLOBAL = CN()
 _C.GLOBAL.HACK = 1.0
+
+# ここから追加
+_C.MODEL.ROI_HEADS.LOSS_TYPE = "bce"  # "focal"または"bce"も選択可能
+_C.MODEL.ROI_HEADS.FOCAL_LOSS_GAMMA = 2.0
+_C.MODEL.ROI_HEADS.FOCAL_LOSS_ALPHA = 0.25