Update README.md

README.md

@@ -96,7 +96,7 @@ pipeline_tag: image-segmentation
   <ul style="list-style-type:disc;">
     <li>Trained with the FLAIR-INC dataset</li>
     <li>RGB images (true colours)</li>
-    <li>U-Net with a mitb5 encoder</li>
+    <li>U-Net with a mit-b5 encoder</li>
     <li>15 class nomenclature : [building, pervious surface, impervious surface, bare soil, water, coniferous, deciduous, brushwood, vineyard, herbaceous, agricultural land, plowed land, swimming pool, snow, greenhouse]</li>
   </ul>
 </div>
@@ -193,7 +193,7 @@ Statistics of the TRAIN+VALIDATION set :
 #### Training Hyperparameters
 
 * Model architecture: Unet (implementation from the [Segmentation Models Pytorch library](https://segmentation-modelspytorch.readthedocs.io/en/latest/docs/api.html#unet))
-* Encoder : Resnet-34 pre-trained with ImageNet
+* Encoder : mit-b5 pre-trained with ImageNet
 * Augmentation :
   * VerticalFlip(p=0.5)
   * HorizontalFlip(p=0.5)
@@ -264,7 +264,7 @@ The following table give the class-wise metrics :
 |  plowed_land   	    |   42.202   |   59.355   |   65.114   |   54.532   |
 |  swimming_pool   	    |   0.000    |   0.000    |   0.000   |   0.000   |
 |  snow   		        |   _0.000_   |   _0.000_   |   _0.000_   |   _0.000_   |
-|  greenhouse   	    |   60.884   |   75.687   |   66.62   |   87.609   |
+|  greenhouse   	    |   60.884   |   75.687   |   66.620   |   87.609   |
 | **average**             |  **53.440**    |  **65.146**  |  **65.517**  |  **65.644**  |
 
 
@@ -278,9 +278,9 @@ The following illustration gives the resulting confusion matrix :
 
 <div style="position: relative; text-align: center;">
     <p style="margin: 0;">Normalized Confusion Matrix (precision)</p>
-    <img src="figs/FLAIR-INC_RVBIE_resnet34_unet_15cl_norm_cm-precision.png" alt="drawing" style="width: 70%; display: block; margin: 0 auto;"/>
+    <img src="FLAIR-INC_rgb_15cl_mitb5-unet_confmat_norm-precision.png" alt="drawing" style="width: 70%; display: block; margin: 0 auto;"/>
     <p style="margin: 0;">Normalized Confusion Matrix (recall)</p>
-    <img src="figs/FLAIR-INC_RVBIE_resnet34_unet_15cl_norm_cm-recall.png" alt="drawing" style="width: 70%; display: block; margin: 0 auto;"/>
+    <img src="FLAIR-INC_rgb_15cl_mitb5-unet_confmat_norm-recall.png" alt="drawing" style="width: 70%; display: block; margin: 0 auto;"/>
 </div>