Datawhale AI 夏令营（第五期）从零上手CV竞赛 Task2：建模方案与上分进阶

一、增加训练的数据集

增加更多的视频到训练集中进行训练，我们将原来的5条视频增加到15条了。

二、增加验证的数据集

增加更多的视频到验证集中以便选取更加优化的模型，我们将原来的3条视频增加到6条了。

三、更换不同的模型预训练权重

 

在原始 notebook 中修改，下载预训练模型部分，以及代码中加载模型部分。

原始为 !wget http://mirror.coggle.club/yolo/yolov8n-v8.2.0.pt -O yolov8n.pt

下面任意一个

!wget http://mirror.coggle.club/yolo/yolov8s-v8.2.0.pt -O yolov8s.pt

!wget http://mirror.coggle.club/yolo/yolov8m-v8.2.0.pt -O yolov8m.pt

!wget http://mirror.coggle.club/yolo/yolov8l-v8.2.0.pt -O yolov8l.pt

!wget http://mirror.coggle.club/yolo/yolov8x-v8.2.0.pt -O yolov8x.pt

最终的结果显示预测的分数有不少提高！