YOLOv5相关三个小技巧

YOLOv5相关三个小技巧
1.训练自己的数据运行改变输出检测框颜色、粗细、标签颜色
utils.plots.py中修改检测框颜色和标签文字颜色
YOLOv5相关三个小技巧_第1张图片
YOLOv5相关三个小技巧_第2张图片
YOLOv5相关三个小技巧_第3张图片
2.输出网络的相关参数
包括所有层数量, 模型总参数量, 需要求梯度的总参数量, img_size大小的mo

你可能感兴趣的:(目标检测模型相关技巧,pytorch,深度学习,神经网络)