YOLOv5相关三个小技巧
YOLOv5相关三个小技巧1.训练自己的数据运行改变输出检测框颜色、粗细、标签颜色utils.plots.py中修改检测框颜色和标签文字颜色2.输出网络的相关参数包括所有层数量, 模型总参数量, 需要求梯度的总参数量, img_size大小的model的浮点计算量GFLOPs直接运行models.yolo.py即可输出相关参数(通过torch_utils.py中的model_info(调用torch_utils.py中的profile)输出模型的相关参数)3.利用摄像头检测相关目标程序
原创
2021-10-20 15:22:46 ·
2922 阅读 ·
2 评论