Closed sherrylsin20 closed 2 years ago
@sherrylsin20,
Can you share the saved model and tflite file with us to reproduce the issue? Thanks!
@sanatmpa1 here is the link for the models: https://drive.google.com/drive/folders/1fkbBzg8lnPI8Hd7hNPk9S_WgHx2axge_?usp=sharing
You can refer to this example document from Tensorflow tflite
to check all the steps.
This issue has been automatically marked as stale because it has no recent activity. It will be closed if no further activity occurs. Thank you.
Closing as stale. Please reopen if you'd like to work on this further.
1. System information
2. Code
Provide code to help us reproduce your issues using one of the following options: Trained my model with model_main_tf2.py script before exporting to saved model using the exporter_main_v2.py script. After that I uploaded my saved model to Google Drive before trying to convert the model to TFLite format. Here is the code for TFLite Convert:
Reference for making the custom object detection model:
1) Reference Object Detection Tutorial: Installation until exporting to saved model
3. Failure after conversion
If the conversion is successful, but the generated model is wrong, then state what is wrong: It changed the output shapes of each outputs from this:
to this:
As you can see, the output shapes of StatefulPartitionedCall 0 - 4 changed. Any solution? I've searched around but I haven't seen any problems similar with the problems I have.
4. Information
Pre trained models used: ssd_mobilenet_v2_fpnlite_640x640_coco17_tpu-8
The only things I've changed from the original pipeline was these: num_classes: 100 (because I have 100 classes) batch_size: 2 deleted all augmentation steps (flip and crop) fine_tune_checkpoint="my_path/pre-trained-models/ssd_mobilenet_v2_fpnlite_640x640_coco17_tpu-8/checkpoint/ckpt-0" num_steps = 100000 fine_tune_checkpoint_type= "detection" use_bfloat16= false train_input_reader=changed the path eval_input_reader=changed the path
I'm sorry if my explanation seems confusing, this is the second time I've ever opened an issue in Github. If you need the saved model and tflite, I'll be willing to share it. Please help!