
(笔记)ubuntu20.04下 yolov5学习与使用
等到数据训练好了以后,就会在主目录下产生一个run文件夹,在run/train/exp/weights目录下会产生两个权重文件,一个是最后一轮的权重文件,一个是最好的权重文件,一会我们就要利用这个最好的权重文件来做推理测试。对视频进行测试,和如上的图片的测试是一样的,只不过是将图片的路径改为视频的路径而已。data:配置文件,训练集、测试集、验证集的路径 还包含目标检测的种类数及种类的名称,和官方提供测试的图片. 如要自己训练数据集是要修改其中的yaml文件。以下是划分数据集的代码。





