关于yolo文件之间的一些关系

Question

在opencv中使用yolo训练出来的模型是pt文件，但是在opencv中使用readNetFromDarknet用到的文件是weight文件，请问一下这两个文件是什么关系呢？然后在该函数内的cfg文件又是如何过去呢？谢谢各位解答

Answer 1

YOLO（You Only Look Once）目标检测算法在训练过程中需要cfg文件和weight文件。

• cfg文件是配置文件，包含了网络结构的信息，如层数、每层神经元的数量等，在训练过程中用来指导网络结构。
• weight文件是权重文件，包含了训练得到的权重和偏置，用来调整网络的参数以提高网络的准确性。

在opencv中使用readNetFromDarknet读取YOLO模型时，需要传入cfg文件和weight文件的路径。而pt文件是pytorch模型，与darknet的weight文件不是同一种格式，需要进行转换。

---

参考资料：

• https://blog.csdn.net/weixin_45067190/article/details/126152960
• https://blog.csdn.net/qinchang1/article/details/89435023
• https://pyimagesearch.com/2018/11/12/yolo-object-detection-with-opencv/