Transfer Learning Semantic Segmentation

[DaveBGld]

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!!

[Peh_Intel]

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]

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