Fuente: Wipo "precision agriculture" Existing segmentation models have the disadvantage that they cannot be deployed on edge devices as they are not compact and require large space. Embodiments disclosed herein provide a method and system for generating a class focused segmentation model using a two stage model training approach, which enables generation of a compact, task specific model. Using a first stage of training, a Machine Learning (ML) segmentation model that is trained on a master training dataset is generated. Further, in a second stage of training, a fine-tuned task specific segmentation model is generated, which can be used for task specific class segmentation. The fine-tuned task specific segmentation model being task specific and in turn compact, can be used edge device deployment and such applications.