Datawhale AI训练营 cvTask2学习心得

经过了Task1的学习,对于cv方向的深度学习有了一定了解,再加上看了训练营的直播经过学长对于比赛赛题的讲解和代码的介绍,让我更好的理解和阅读代码。在最开始看到代码实际上是有一点对于未知的恐惧的,不过经过学长细心地讲解已经理解了整个baseline的代码逻辑和大体架构。
我在看完Task2的打卡文件之后,对照文中给出的修改建议对代码进行了修改,并在云服务器上进行了运行。我将训练所用的模型换成了YOLO8s,并且根据更新的代码仓库中针对YOLO进阶代码,对之前给出的baseline中的代码进行了修改。

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

import warnings
warnings.filterwarnings('ignore')

from ultralytics import YOLO
model = YOLO("yolov8s.pt")
results = model.train(data="yolo-dataset/yolo.yaml", epochs=30, batch=16)

经过修改后所得出的结果比最开始的代码得分提高了0.3,由0.4提升到了0.7左右,虽然还是一般都是对于我这个没什么深度学习代码基础的人来说,已经很不错了。接下来打算对代码再进行一些其他的修改以期待分数可以提高更多,希望可以在接下来的比赛中有所成就。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值