How to convert the Mask RCNN model to tflite model

gustavz / realtime_object_detection

Plug and Play Real-Time Object Detection App with Tensorflow and OpenCV

MIT License

281 stars 148 forks source link

How to convert the Mask RCNN model to tflite model #41

Closed noah003 closed 5 years ago

noah003 commented 5 years ago

Have you converted Mask_RNN model to .tflite successfully? when I run the script as shown in scripts/create_tflite_graph.sh, I got the following error

tensorflow/lite/toco/tooling_util.cc:908] Check failed: GetOpWithOutput(model, output_array) Specified output array "num_detections,detection_boxes,detection_scores,detection_classes,detection_masks" is not produced by any op in this graph. Is it a typo? To silence this message, pass this flag:  allow_nonexistent_arrays

gustavz commented 5 years ago

This repo is not about Mask R-CNN.

And no unfortunately as for my knowledge the mask r-cnn layers are not yet fully supported in tflite