bigmlcom / sensenet

0 stars 8 forks source link

TF Upgrade / Feature extraction wrapper #36

Closed charleslparker closed 2 years ago

charleslparker commented 2 years ago

This (mostly) adds a function models.wrappers.create_image_feature_extractor that will return model object that "predicts the image features" instead of the classes (see the documentation). You can either pass in a BigML image deepnet or the name of a pretrained model (e.g., "resnet18").

/cc @unmonoqueteclea @aclementev