yolo.py的主要作用是构建yolov5的模型;
而且这个yolo.py文件可以单独执行;
这里主要对目标检测中的相关类进行了注释解读,分割等没有用到的暂时没有注释。
第一部分:导入包,配置路径等
第二部分:程序入口,执行程序;在这部分中,创建了一张640×640×3的图片测试,送入Model类中进行了测试。
第三部分:Model类的执行;这里Model又根据DetectionModel创建而来,因此分析DetectionModel即可
第四部分:在DetectionModel中,执行了parse_model类。
这里parse_model进行了yolov5s.yaml(不同yaml文件的解析)
# YOLOv5
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)