Remove ImplicitKeypointBBoxHead (#98)

luxonis · Oct 7, 2024 · 157b44d · 157b44d
1 parent 4c90b8b
commit 157b44d
Show file tree

Hide file tree

Showing 21 changed files with 3 additions and 1,234 deletions.
diff --git a/luxonis_train/attached_modules/losses/__init__.py b/luxonis_train/attached_modules/losses/__init__.py
@@ -3,8 +3,6 @@
 from .bce_with_logits import BCEWithLogitsLoss
 from .cross_entropy import CrossEntropyLoss
 from .efficient_keypoint_bbox_loss import EfficientKeypointBBoxLoss
-from .implicit_keypoint_bbox_loss import ImplicitKeypointBBoxLoss
-from .keypoint_loss import KeypointLoss
 from .sigmoid_focal_loss import SigmoidFocalLoss
 from .smooth_bce_with_logits import SmoothBCEWithLogitsLoss
 from .softmax_focal_loss import SoftmaxFocalLoss
@@ -14,8 +12,6 @@
     "BCEWithLogitsLoss",
     "CrossEntropyLoss",
     "EfficientKeypointBBoxLoss",
-    "ImplicitKeypointBBoxLoss",
-    "KeypointLoss",
     "BaseLoss",
     "SigmoidFocalLoss",
     "SmoothBCEWithLogitsLoss",

diff --git a/luxonis_train/attached_modules/losses/implicit_keypoint_bbox_loss.py b/luxonis_train/attached_modules/losses/implicit_keypoint_bbox_loss.py