search

onnx / keras-onnx

Convert tf.keras/Keras models to ONNX
Apache License 2.0
379 stars 110 forks source link

How to convert resnet101-based Mask-RCNN model to ONNX ? #716

Open pra-dan opened 3 years ago

pra-dan commented 3 years ago

Hi,

I have my my Mask-RCNN model weights (h5) and its based on ResNet101. I am unable to load my weights before converting them to ONNX. The example you provide, loads a ResNet50 weight 

# load keras model
from keras.applications.resnet50 import ResNet50
model = ResNet50(include_top=True, weights='imagenet')

There is no such utility for resnet101. How should I go forward ?