改动yolov5转化为tflite

版本6.1,修改了yolov5的部分网络结构,使用模型转换代码时出错,发现原因是tf.py中没有对应的代码结构。请教大佬们common.py中的代码如何对应写道tf.py中?export.py要做哪些相应改动呢?

用tensorflow实现一下啊,这里面参考下其他网络层改写就行
举个例子来说,都是C3模块的不同框架实现

img


img

最后我估计你还要改下tf.py里面的parse_model这个函数,这个是用来建网络结构的,如果你已经训练的话,这部分应该已经该过了。

如果你不提供代码,所有的需求仅仅让大家猜测你遇到的情况,爱晚乏客游的例子很好,你要做的是查找tf.py中和common.py中修改的网络结构相应的代码段,并将其修改为与TensorFlow兼容的代码。如果涉及到tensflow的部分,还会有变化。