ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix: inverted logic

Browse source 
Former-commit-id: b3a388a527875cdc65ee3ee5c1383181bc75b6c6 [formerly 8b53736ce5adf77812831c7a3fdbca8b3b376edb]
Former-commit-id: 35b5ad1b73d5f3682fdd4251e280706373d17d36
This commit is contained in:
Laurent Fainsin 2022-07-08 13:35:08 +02:00
parent 6acac70972
commit 1a94fa6c09
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/unet/model.py
View file

@ -195,7 +195,7 @@ class UNet(pl.LightningModule):
accuracy = (masks_true == masks_pred_bin).float().mean() accuracy = (masks_true == masks_pred_bin).float().mean()
rows = [] rows = []
if batch_idx % 50 == 0 or dice < 0.1: if batch_idx % 50 == 0 or dice > 0.9:
for i, (img, mask, pred, pred_bin) in enumerate( for i, (img, mask, pred, pred_bin) in enumerate(
zip( zip(
images.cpu(), images.cpu(),