当前位置:   article > 正文

YoloV5源码部分注释解读(ultralytics版本)(yolo.py)_yolov5 ultralytics.yolo版本

yolov5 ultralytics.yolo版本

yolo.py的主要作用是构建yolov5的模型;

而且这个yolo.py文件可以单独执行;

这里主要对目标检测中的相关类进行了注释解读,分割等没有用到的暂时没有注释。

第一部分:导入包,配置路径等
第二部分:程序入口,执行程序;在这部分中,创建了一张640×640×3的图片测试,送入Model类中进行了测试。
第三部分:Model类的执行;这里Model又根据DetectionModel创建而来,因此分析DetectionModel即可
第四部分:在DetectionModel中,执行了parse_model类。
这里parse_model进行了yolov5s.yaml(不同yaml文件的解析)

# YOLOv5 声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签