Feature Extractor Regression for other Models

Hi there,

I was watching the online tutorials on feature extraction and regression and noticed they were all done on the MobileNet model. However, I'm trying to do something similar but with a input from the sketch canvas, much like the DoodleNet model. I tried implementing it but can't seem to get it to work.

Is there any way of implementing the feature extractor regression for custom models or one such as DoodleNet?