File tree 2 files changed +10
-5
lines changed
segmentation_models_pytorch/losses
2 files changed +10
-5
lines changed Original file line number Diff line number Diff line change @@ -5,6 +5,6 @@ pillow==8.0.0
5
5
pretrainedmodels==0.7.1
6
6
six==1.5.0
7
7
timm==0.9.0
8
- torch==1.8 .0
9
- torchvision==0.9 .0
8
+ torch==1.9 .0
9
+ torchvision==0.10 .0
10
10
tqdm==4.42.1
Original file line number Diff line number Diff line change @@ -192,9 +192,14 @@ def soft_tversky_score(
192
192
"""
193
193
assert output .size () == target .size ()
194
194
195
- output_sum = torch .sum (output , dim = dims )
196
- target_sum = torch .sum (target , dim = dims )
197
- difference = LA .vector_norm (output - target , ord = 1 , dim = dims )
195
+ if dims is not None :
196
+ output_sum = torch .sum (output , dim = dims )
197
+ target_sum = torch .sum (target , dim = dims )
198
+ difference = LA .vector_norm (output - target , ord = 1 , dim = dims )
199
+ else :
200
+ output_sum = torch .sum (output )
201
+ target_sum = torch .sum (target )
202
+ difference = LA .vector_norm (output - target , ord = 1 )
198
203
199
204
intersection = (output_sum + target_sum - difference ) / 2 # TP
200
205
fp = output_sum - intersection
You can’t perform that action at this time.
0 commit comments