如何查看YOLO网络结构

当添加新的结构模块时候,我们常常需要修改并查看新生成的网络结构。
例如,在YOLO算法改进4【中阶改进篇】中我们添加了DCnv2的可变型卷积。

YOLO算法改进4【中阶改进篇】:添加DeformableConvolution v2可变形卷积模块
文章地址:https://blog.csdn.net/qq_39522016/article/details/134150391

运行以下命令执行打印网络结构:

python models/yolo.py --cfg models/yolov5s.yaml

你可能感兴趣的:(Yolov5问题解决,YOLO)