No module named 'models'

[jhnhp]

I tried this code to convert my custom trained YOLOv5 model (.pt)
It returns this message

2021-05-31 10:26:03.181164: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0
No module named 'models'

How can I fix this?

[Linxin-Zhong]

I got the same problem and don't know how to fix this either

[mozeqiu]

I got the same problem and don't know how to fix this either

I suggest to try the steps separately

[thongvhoang]

I got the same problem and don't know how to fix this either

[codebugged]

We are waiting @omerferhatt

[mozeqiu]

我们在等@omerferhatt

I have successfully converted, I will proceed in separate steps。like pt2onnx.py --> onnx2tf.py ---> tf2tflite.py,
Instead of finishing the conversion

[omerferhatt]

Hey everyone, sorry for late respond. I change code a little bit. Please test it out and lemme now if anything is broken.

[weinixuehao]

No module named 'model'