diff --git a/bob/ip/binseg/modeling/driubn.py b/bob/ip/binseg/modeling/driubn.py
index 4c70dc4bc13c1a1f4101fa16fba64a4ff9d66e70..6043fcd6257b73712572002d1895fca84ca7e69f 100644
--- a/bob/ip/binseg/modeling/driubn.py
+++ b/bob/ip/binseg/modeling/driubn.py
@@ -74,7 +74,7 @@ def build_driu():
-------
:py:class:torch.nn.Module
"""
- backbone = vgg16_bn(pretrained=False, return_features = [3, 8, 14, 22])
+ backbone = vgg16_bn(pretrained=False, return_features = [5, 12, 19, 29])
driu_head = DRIU([64, 128, 256, 512])
model = nn.Sequential(OrderedDict([("backbone", backbone), ("head", driu_head)]))
diff --git a/doc/configs.rst b/doc/configs.rst
index 85bbbe4de39d2e94b5b260736c78ce300a1a7231..51814cfc25432a1643b62bce95280b07798e53b1 100644
--- a/doc/configs.rst
+++ b/doc/configs.rst
@@ -175,6 +175,12 @@ DRIUSSL
--------
.. literalinclude:: ../bob/ip/binseg/configs/models/driussl.py
+.. _bob.ip.binseg.configs.models.driubnssl:
+
+DRIUBNSSL
+---------
+.. literalinclude:: ../bob/ip/binseg/configs/models/driubnssl.py
+
.. _bob.ip.binseg.configs.models.m2unetssl:
diff --git a/doc/covdresults.rst b/doc/covdresults.rst
index 94ed0ab7c3871e8e985abe01896267f9a9511acc..9148f1ba2661d6ebdb4580b8fb6ea3f02c7fd3f1 100644
--- a/doc/covdresults.rst
+++ b/doc/covdresults.rst
@@ -9,27 +9,29 @@ COVD- and COVD-SLL Results
F1 Scores
===========
-+-------------------+---------------+---------+
-| | DRIU | M2U-Net |
-+-------------------+---------------+---------+
-| COVD-DRIVE | 0.7896 | 0.7906 |
-+-------------------+---------------+---------+
-| COVD-DRIVE SSL | 0.7870 | 0.7938 |
-+-------------------+---------------+---------+
-| COVD-STARE | 0.7979 | 0.8120 |
-+-------------------+---------------+---------+
-| COVD-STARE SSL | 0.8062 | 0.8222 |
-+-------------------+---------------+---------+
-| COVD-CHASEDB1 | 0.7979 | 0.7898 |
-+-------------------+---------------+---------+
-| COVD-CHASEDB1 SSL | 0.7976 | 0.8000 |
-+-------------------+---------------+---------+
-| COVD-HRF | 0.8013 | 0.8036 |
-+-------------------+---------------+---------+
-| COVD-HRF SSL | 0.8002 | 0.7999 |
-+-------------------+---------------+---------+
-| COVD-IOSTAR | 0.7934 | 0.7953 |
-+-------------------+---------------+---------+
-| COVD-IOSTAR SSL | 0.7995 | 0.7868 |
-+-------------------+---------------+---------+
+F1 score together with standard deviation across test images.
+
++-------------------+--------+-----------------+---------+
+| F1 score | DRIU | DRIU BN | M2U-Net |
++-------------------+--------+-----------------+---------+
+| COVD-DRIVE | 0.7896 | 0.8000 (0.0182) | 0.7906 |
++-------------------+--------+-----------------+---------+
+| COVD-DRIVE SSL | 0.7870 | 0.8020 (0.0179) | 0.7938 |
++-------------------+--------+-----------------+---------+
+| COVD-STARE | 0.7979 | 0.8129 (0.0986) | 0.8120 |
++-------------------+--------+-----------------+---------+
+| COVD-STARE SSL | 0.8062 | 0.8221 (0.0784) | 0.8222 |
++-------------------+--------+-----------------+---------+
+| COVD-CHASEDB1 | 0.7979 | 0.7923 (0.0240) | 0.7898 |
++-------------------+--------+-----------------+---------+
+| COVD-CHASEDB1 SSL | 0.7976 | 0.7992 (0.0235) | 0.8000 |
++-------------------+--------+-----------------+---------+
+| COVD-HRF | 0.8013 | 0.8027 (0.0452) | 0.8036 |
++-------------------+--------+-----------------+---------+
+| COVD-HRF SSL | 0.8002 | | 0.7999 |
++-------------------+--------+-----------------+---------+
+| COVD-IOSTAR | 0.7934 | 0.7763 (0.0311) | 0.7953 |
++-------------------+--------+-----------------+---------+
+| COVD-IOSTAR SSL | 0.7995 | 0.7904 (0.0215) | 0.7868 |
++-------------------+--------+-----------------+---------+
diff --git a/doc/evaluation.rst b/doc/evaluation.rst
index 3c22e88e0bcc60410ec112750da02761fea30ee7..6ffbf673e56b1fbbfa8b7e01db77c2537a51787c 100644
--- a/doc/evaluation.rst
+++ b/doc/evaluation.rst
@@ -31,14 +31,14 @@ Pretrained Models
=================
Due to storage limitations we only provide weights of a subset
-of all evaluated models, namely all DRIU and M2U-Net variants:
+of all evaluated models:
+--------------------+--------------------+--------------------------------------------------------------------------------------------------------------------+
| | DRIU | M2UNet |
+--------------------+--------------------+--------------------------------------------------------------------------------------------------------------------+
-| DRIVE | `DRIU_DRIVE.pth`_ | |
+| DRIVE | `DRIU_DRIVE.pth`_ | `M2UNet_DRIVE.pth <m2unet_drive.pth_>`_ |
+--------------------+--------------------+--------------------------------------------------------------------------------------------------------------------+
| COVD-DRIVE | | `M2UNet_COVD-DRIVE.pth <https://dl.dropboxusercontent.com/s/x5wb84uao8nlx44/M2UNet_COVD-DRIVE.pth>`_ |
+--------------------+--------------------+--------------------------------------------------------------------------------------------------------------------+
diff --git a/doc/training.rst b/doc/training.rst
index f756524f190a4656d91f22bfca0b4f23fbeca54e..5e5c83a4ada67722c274039429bd4731d844c72c 100644
--- a/doc/training.rst
+++ b/doc/training.rst
@@ -63,7 +63,9 @@ Default Model configs
=====================
* DRIU
+* DRIUBN
* DRIUSSL
+* DRIUBNSSL
* DRIUOD
* HED
* M2UNet
@@ -205,7 +207,7 @@ COVD-:
# model configs
driu=DRIU
m2u=M2UNet
- b_driu=4
+ b_driu=2
b_m2u=4
# Train
bob binseg train $driu $dataset -b $b_driu -d cuda -o $output"/"$driu -vv
@@ -218,7 +220,7 @@ COVD-:
# model configs
driu=DRIU
m2u=M2UNet
- b_driu=4
+ b_driu=2
b_m2u=4
# Train
bob binseg train $driu $dataset -b $b_driu -d cuda -o $output"/"$driu -vv
@@ -288,7 +290,7 @@ COVD-SSL:
# model configs
driu=DRIUSSL
m2u=M2UNetSSL
- b_driu=1
+ b_driu=2
b_m2u=2
# Train
bob binseg ssltrain $driu $dataset -b $b_driu -d cuda -o $output"/"$driu -vv