From 0fd37da989930651e7ec653c7b6c16ee7747b71b Mon Sep 17 00:00:00 2001
From: ricky0614 <162050908+ricky0614@users.noreply.github.com>
Date: Mon, 23 Dec 2024 12:19:14 +0900
Subject: [PATCH] Update defaults.py

add default loss function
---
 detectron2/config/defaults.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/detectron2/config/defaults.py b/detectron2/config/defaults.py
index 506651730e..3c72580e58 100644
--- 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