模型转换 - Githubissues

bubbliiiing / yolov4-tf2

这是一个yolo4-tf2（tensorflow2）的源码，可以用于训练自己的模型。

MIT License

314 stars 111 forks source link

Open NJR-alex opened 3 years ago

NJR-alex commented 3 years ago

想问一下训练出来的.h5文件有办法转换成.weights格式吗，或者如何修改save的方式，直接保存成.weights，谢谢！

bubbliiiing commented 3 years ago

我没有办法，不过那个不是darknet的格式吗，为何直接用那个库训练

NJR-alex commented 3 years ago

我没有办法，不过那个不是darknet的格式吗，为何直接用那个库训练

darknet训练效果一直不咋地，不清楚问题出在哪里，对远处目标的检测效果比较差，所以想在TF下训练再转成.weights格式，提供给工程使用。

bubbliiiing commented 3 years ago

啊这，我确实没有转换的代码

6Lackiu commented 3 years ago

我没有办法，不过那个不是darknet的格式吗，为什么直接用那个库训练

darknet训练效果一直不咋地，明显问题出在哪里，对远处目标的检测效果比较差，所以想在TF下训练再转成.weights格式，提供给工程使用。

我也想知道这个问题，请问您解决了吗？

muzzziy commented 3 years ago