2024年1月24日
yolov5训练时参数workers与batch-size的区别常见问题
摘要： 很多新手在使用yolov5训练模型的时候会出现爆内存和爆显存问题，一般就是由于worker和batch_size参数设置过大有关，参考如下是解决的方法。 一、workers train.py中关于workers设置代码如下： workers是指数据装载时cpu所使用的线程数，默认为8，但是按照默认的 阅读全文
posted @ 2024-01-24 16:15 txwtech 阅读(971) 评论(1) 推荐(0) 编辑
labelimg选择标注格式yolo
摘要： 翻译 搜索 复制 阅读全文
posted @ 2024-01-24 11:42 txwtech 阅读(21) 评论(1) 推荐(0) 编辑