tensorflow / tensorflow/models
Allow add_postprocessing_op=false option in export_tflite_graph_tf2.py
Open
Nobody has claimed this yet.
models:research:odapi
type:feature
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
In file:
object_detection/export_tflite_graph_tf2.py
There is no option to remove postprocessing ops (--add_postprocessing_op=false) to support feeding the model directly into MediaPipe which only needs the raw box encodings and class predictions as stated here:
Previous export_tflite_ssd_graph.py script had this option. It would be nice to have the same option in the tf2 script or if there anyway to remove postprocessing from the model it would be helpful to know how.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.