tensorflow / tensorflow/models

How to use Tensorflow Object Detection API with multi-channel images

Open
#9,044 2 comments 1 reaction 3 assignees View on GitHub

@pkulzc is already working on this.

Since Aug 5, 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.

  • [√] 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/r1.13.0/research/object_detection

2. Describe the feature you request

I work with multi-channel images (5 channels). I'm trying to train an object detection model with this images but I haven't found any good support to do it.

I attempt to train a faster RCNN inception resnet v2 network.

3. Additional context

To be more specific, I'm able to modify the input layer ([3, 3, 3, 32] to [3, 3, 5, 32]) in order to adapt one pre-trained model and I'm able to export this model and perform inference with it.
For this, I made modifications based on this: https://github.com/minhnhat93/tf_object_detection_multi_channels

The problem comes when I want to train a model using multiple-channel images.

I've tried using the --num_additional_channels option in model_main.py, but I couldn't make it work.

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.