Prerequisites

Please answer the following questions for yourself before submitting an issue.

• [x] I am using the latest TensorFlow Model Garden release and TensorFlow 2.
• [x] I am reporting the issue to the correct repository. (Model Garden official or research directory)
• [x] I checked to make sure that this issue has not been filed already.

1. The entire URL of the file you are using

https://github.com/tensorflow/models/blob/master/research/delf/delf/python/training/README.md

2. Describe the bug

1. Successful generating tfrecords with build_image_dataset.py script
2. Get error when used train.py as instruction said

3. Steps to reproduce

Use train.py as instruction said

4. Expected behavior

There is a check of all images in build_image_dataset, so all of my dataset is correct. How is possible that channel numbers in image is mismatched with expected input shape?

5. Additional context

Cannot add tensor to the batch: number of elements does not match. Shapes are: [tensor]: [321,321,1], [batch]: [321,321,3]