tensorflow / tensorflow/models

Exporting object detection inference graph using custom input dimensions, with the option to remove preprocessing steps

Open
#8,960 0 comments 0 reactions 3 assignees View on GitHub

@pkulzc is already working on this.

Since Jul 27, 2020.

models:research:odapi type:feature
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

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.