diff --git a/README.md b/README.md index d21040b..6cfb407 100644 --- a/README.md +++ b/README.md @@ -212,8 +212,10 @@ We can see some good segmentation like in demo 1 or demo 2 but it becomes more d # Conclusion/Discussion -Using adapters we partially answered the task of segmenting rings. In the demo examples we saw that the model still struggles to caputre all the rings. This could be because the training dataset is small (8 images). -To improve the model, we can add checkpoint and early stopping with a validation set to prevent from overfitting. In addition, doing data augmentation could be interesting to strengthen the model generalization. +Using adapters, I partially succeded the task of segmenting rings. In the demo examples we saw that the model still struggles to caputre all the rings. This could be because the training dataset is small (only 8 images). +To improve the model, we can train on higher number of epochs, add model checkpoint and early stopping with a validation set to prevent from overfitting. In addition, doing data augmentation could be interesting to strengthen the model generalization. + +Another interesting idea could be to apply LoRA to other blocks of SAM like the mask decoder which is smaller than the image encoder. This could save training time and potentially show better results. # Folder layout .