Intel® Distribution of OpenVINO™ Toolkit
Community support and discussions about the Intel® Distribution of OpenVINO™ toolkit, OpenCV, and all things computer vision-related on Intel® platforms.

Transfer Learning Semantic Segmentation

DaveBGld
Beginner
190 Views

New to OpenVino.

 

Looking to do transfer learning for a semantic segmentation application that has a class with few pixels, resulting in class imbalance.

 

found this video, but the links to opencv/training_extensions at github is not there...

https://www.youtube.com/watch?v=wmRNqg_7Eo0 

 

Found this repo https://github.com/openvinotoolkit/training_extensions but is seems to have less info than what the video shows.

 

Looking for a guide to

 

a) find pretrained models that can be used for transfer learning,

and

b) A guide to use training extensions for retraining and pruning and optimizing with my custo dataset.

 

Thanks!!

 

0 Kudos
2 Replies
Peh_Intel
Moderator
152 Views

Hi DaveBGld,


Thanks for reaching out to us.


Please refer to this openvinotoolkit/mmsegmentation repository for more information.


a)      You can view the pre-trained models in the model zoo.

b)     You can refer to the get started guide of MMSegmentation.



Regards,

Peh


Peh_Intel
Moderator
133 Views

Hi DaveBGld,


This thread will no longer be monitored since we have provided answers. If you need any additional information from Intel, please submit a new question. 



Regards,

Peh


Reply