1.配置文件
配置文件(也称为模型定义文件或模型结构文件)包含了模型的结构信息,如层的类型、数量、参数等;
即YOLOV5的框架
图片来自:yolov5-5.0版本(目前最新)网络结构图_yolov5网络结构_昼行plus的博客-CSDN博客
2.权重文件
权重文件(也称为模型参数文件或模型状态文件)包含了模型的参数信息,即模型的每一层中的权重和偏置等参数。
3.YOLOV5相关
在使用 YOLOv5 进行目标检测时,需要将配置文件和权重文件结合起来使用,即先根据配置文件创建模型的结构,然后再将权重文件中的参数加载到模型中,从而得到完整的检测器。这样,检测器才能够正确地识别输入图像中的物体,并输出物体类别、位置、置信度等信息。
在最新的YOLOV5的detect.py中并没有显式指定配置文件的路径,而是根据权重文件推理配置文件,具体代码如下
#detect.py
......
# Load model
device = select_device(device)
model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data, fp16=half)
stride, names, pt = model.stride, model.names, model.pt
imgsz = check_img_size(imgsz, s=stride) # check image size
......
其中的DetectMultiBackend()函数就是此功能。
如有错误,望大佬指出!
版权归原作者 张量Tensor 所有, 如有侵权,请联系我们删除。