Error Converting weights of p7 from pytorch to tensorflow. Could you help in this

wangermeng2021 / Scaled-YOLOv4-tensorflow2

A Tensorflow2.x implementation of Scaled-YOLOv4 as described in Scaled-YOLOv4: Scaling Cross Stage Partial Network

Apache License 2.0

47 stars 18 forks source link

Error Converting weights of p7 from pytorch to tensorflow. Could you help in this #16

Closed Palak-15 closed 2 years ago

Palak-15 commented 2 years ago

I have tried converted .pt model to .onxx and then onxx to pb file for tensorflow. I have checked the graph using neutron => 1 transpose layer has been added so that it can handle NHWC input of tensorflow.

Palak-15 commented 2 years ago

Solved!

tylertroy commented 2 years ago

@Palak-15 How did you solve this?