MASKRCNN TFlite conversion

661 views
Skip to first unread message

saurav solanki

unread,
Jun 11, 2021, 1:14:01 AM6/11/21
to TensorFlow Lite
Hi friends,
I would like to know if anyone has converted MASKRCNN sucessfully to TFlite and taken inference on sample images. After days of struggle and googling, I could not able to find the right way. Any help will be welcome.

Best,
Saurav

Jaesung Chung

unread,
Jun 13, 2021, 10:27:52 PM6/13/21
to TensorFlow Lite, sausol....@gmail.com
Hi,

We have verified the following TF hub model can be converted to the corresponding TFLite model.


You can also look at the TensorFlow model garden to create your own object detection model.


Best regards,
Jaesung

Andy Toan Tran

unread,
Jul 9, 2021, 8:53:00 PM7/9/21
to TensorFlow Lite, Jaesung Chung, sausol....@gmail.com
Hi Jaesung,
I tried converting the  https://tfhub.dev/tensorflow/mask_rcnn/inception_resnet_v2_1024x1024/1 to a TFLite model using the saved model conversion here https://www.tensorflow.org/lite/convert but when I convert it with Tensorflow 2 I get an error and when I use tf-nightly it converts. Is the conversion not possible in TensorFlow2?
Best regards,
Andy

Jaesung Chung

unread,
Jul 11, 2021, 4:49:15 AM7/11/21
to Andy Toan Tran, TensorFlow Lite, sausol....@gmail.com
I would recommend using TensorFlow 2.5 or beyond versions.

2021년 7월 10일 (토) 오전 9:53, Andy Toan Tran <and...@uci.edu>님이 작성:

Andy Toan Tran

unread,
Jul 12, 2021, 2:41:00 PM7/12/21
to TensorFlow Lite, Jaesung Chung, TensorFlow Lite, sausol....@gmail.com, Andy Toan Tran
Hi Jaesung,
I am using TensorFlow 2.5 now and using the same code from the link to convert but it is giving me a segmentation fault now. When I run it using tf-nightly it still converts though. Do you know any reason why?
Best,
Andy

Jaesung Chung

unread,
Jul 12, 2021, 6:07:04 PM7/12/21
to Andy Toan Tran, TensorFlow Lite, sausol....@gmail.com
It is hard to tell the reason without looking at the contexts. Could you file an issue with reproducible steps at the TensorFlow github?

2021년 7월 13일 (화) 오전 3:41, Andy Toan Tran <and...@uci.edu>님이 작성:

Debdeep Goswami

unread,
Jul 15, 2021, 10:55:44 AM7/15/21
to TensorFlow Lite, Jaesung Chung, TensorFlow Lite, sausol....@gmail.com, and...@uci.edu
Hi all, I want to detect number of objects in raspberry pi4 from a video stream. I have used SSD Mobilenet and the maximum number of objects it can detect is 10. I want to use Mask RCNN  but unable to convert it in tflite.. Any help ??
Reply all
Reply to author
Forward
0 new messages