tensorflow / tensorflow/models
Exporting object detection inference graph using custom input dimensions, with the option to remove preprocessing steps
@pkulzc is already working on this.
Since Jul 27, 2020.
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
Prerequisites
Please answer the following question for yourself before submitting an issue.
- [x ] I checked to make sure that this feature has not been requested already.
1. The entire URL of the file you are using
https://github.com/tensorflow/models/tree/master/research/object_detection/...
2. Describe the feature you request
Current implementation of TensorFlow 2 object detection api allows for exporting of a trained models inference graph using only an image tensor, a float image tensor, an encoded image string tensor and tf example string tensor. It would be extremely useful to be able to define custom input tensor dimensions i.e. something which has more than 3 channels, as well as have the option to remove any preprocessing steps such as image resizing. (My custom trained model never encounters any different input dimensions or data types besides float32).
3. Additional context
If anyone has any advice on how to add these features please let me know! And also if there already is an option/method to remove any preprocessing steps when exporting a model please forgive my ignorance. Thank you!!
4. Are you willing to contribute it? (Yes or No)
Yes!!
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.