Update README.md

README.md

@@ -212,26 +212,7 @@ Input normalization was performed
 * Optimizer : SGD
 * Schaeduler : mode = "min", factor = 0.5, patience = 10, cooldown = 4, min_lr = 1e-7
 * Learning rate : 0.02
-* Class Weights : 
-  * 1: [1, building]
-  * 2: [1, pervious surface]
-  * 3: [1, impervious surface]
-  * 4: [1, bare soil]
-  * 5: [1, water]
-  * 6: [1, coniferous]
-  * 7: [1, deciduous]
-  * 8: [1, brushwood]
-  * 9: [1, vineyard]
-  * 10: [1,herbaceous vegetation]
-  * 11: [1, agricultural land]
-  * 12: [1, plowed land]
-  * 13: [1, swimming_pool]
-  * 14: [1, snow]
-  * 15: [0, clear cut]
-  * 16: [0, mixed]
-  * 17: [0, ligneous]
-  * 18: [1, greenhouse]
-  * 19: [0, other]
+* Class Weights : [1-building: 1.0 , 2-pervious surface: 1.0 , 3-impervious surface: 1.0 , 4-bare soil: 1.0 , 5-water: 1.0 , 6-coniferous: 1.0 , 7-deciduous: 1.0 , 8-brushwood: 1.0 , 9-vineyard: 1.0 , 10-herbaceous vegetation: 1.0 , 11-agricultural land: 1.0 , 12-plowed land: 1.0 , 13-swimming_pool: 1.0 , 14-snow: 1.0 , 15-clear cut: 0.0 , 16-mixed: 0.0 , 17-ligneous: 0.0 , 18-greenhouse: 1.0 , 19-other: 0.0]
 
 
 #### Speeds, Sizes, Times [optional]
@@ -242,9 +223,7 @@ The FLAIR-INC_RVBIE_resnet34_unet_15cl_norm model was trained on a HPC/AI resour
 
 FLAIR-INC_RVBIE_resnet34_unet_15cl_norm was obtained for num_epoch=76 with corresponding val_loss=0.56. 
 
-<img src="train_loss_FLAIR-INC_RGBIE_resnet34_unet_15cl_norm.png" alt="drawing" style="width:200px;"/>
-
-
+<!-- <img src="train_loss_FLAIR-INC_RGBIE_resnet34_unet_15cl_norm.png" alt="drawing" style="width:200px;"/>-->
 
 Train loss log             |  Val loss log 
 :-------------------------:|:-------------------------:
@@ -252,12 +231,6 @@ Train loss log             |  Val loss log
 
 
 
-
-Train loss log             |  Val loss log 
-:-------------------------:|:-------------------------:
-<img src="train_loss_FLAIR-INC_RGBIE_resnet34_unet_15cl_norm.png" alt="drawing" style="width:200px;"/> | <img src="val_loss_FLAIR-INC_RGBIE_resnet34_unet_15cl_norm.png" alt="drawing" style="width:200px;"/>
-
-
 {{ speeds_sizes_times | default("[More Information Needed]", true)}}
 
 ## Evaluation