Skip to content

Commit

Permalink
Add equation
Browse files Browse the repository at this point in the history
  • Loading branch information
mrokuss committed Jul 10, 2024
1 parent 057de34 commit cb5dadd
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,9 @@ In the code the Tubed Skeletonization is done during *dataloading*, see the [cod

#### Full Loss calculation:

$$\mathcal{L} = \mathcal{L}_{Dice} + \mathcal{L}_{CE} + w \cdot \mathcal{L}_{SkelRecall}$$
```math
\mathcal{L} = \mathcal{L}_{Dice} + \mathcal{L}_{CE} + w \cdot \mathcal{L}_{SkelRecall}
```

You can change the weight of the additional Skeleton Recall Loss term by modifying the value of `self.weight_srec` in the [nnUNetTrainerSkeletonRecall](nnunetv2/training/nnUNetTrainer/variants/loss/nnUNetTrainerSkeletonRecall.py)

Expand Down

0 comments on commit cb5dadd

Please sign in to comment.