Skip to content

DirectML returning empty result with ObjectDetection (Mobilinet V2 FPN Keras) #2325

Open
@willianwrm

Description

@willianwrm

Describe the bug
Result is empty when using DirectML but works as expected when using only CPU.
Tested for ObjectDection with ssd_mobilenet_v2_fpn_keras model.

Urgency
low

System information

  • OS Platform and Distribution: Windows 11 23H2 22631.3447
  • ONNX version: 1.17.3
  • GPU: NVIDIA RTX 3060Ti driver version 31.0.15.5212 and Intel UHD 730 driver version 31.0.101.5084

System information for training and ONNX model conversion

  • OS Platform and Distribution: Linux Ubuntu 22.04.3 LTS in WSL2
  • TensorFlow: 2.15
  • ONNX version: 1.16.1, tested with GIT version as well
  • piplist.txt

To Reproduce
Use this C# .NET 8.0 Windows Forms App, select a file image and wait for the result, the CPU result is printed in blue and DirectML result in red.
The project contains the onnx model and image examples can be found here. Tested opset 18 to 13, no change.

Screenshots
image

Additional context
Training configuration, data set, command lines and tensorflow exported graph can be found here

Metadata

Metadata

Assignees

Labels

bugAn unexpected problem or unintended behaviorpending on user responseWaiting for more information or validation from user

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions