keras yolo8目标检测

是从coco数据集提取其中的veh_ids=[3,6,8,10]
labels=['car','bus','truck','traffic light']来做目标检测,分别表示汽车,公交车,卡车,交通灯,用的backbone = keras_cv.models.YOLOV8Backbone.from_preset(
    "yolo_v8_m_backbone_coco"  
),不用预训练权重非常慢,训练过程如下,先是在8000个样本上训练的,淡蓝的是真实框,黄色的是模型预测,下面的是训练几个轮次后的情况

 上面是在从coco提取的4个种类,之前没在所有数据集上训练,现在在所有数据集上训练

7915/7915 [==============================] - 2097s 265ms/step - loss: 1.6061 - box_loss: 1.3257 - class_loss: 0.2804 - val_loss: 1.6310 - val_box_loss: 1.3246 - val_class_loss: 0.3064 - val_AP: 0.1829 - val_AP50: 0.2804 - val_AP75: 0.1981 - val_APs: 0.0474 - val_APm: 0.2140 - val_APl: 0.2302 - val_ARmax1: 0.2741 - val_ARmax10: 0.4171 - val_ARmax100: 0.4198 - val_ARs: 0.1535 - val_ARm: 0.4595 - val_ARl: 0.4680
 目前损失指标是这样,效果还不错吧!之前我加了火车也是检测目标,但是火车太大,一般没人要求把火车当目标,所以掉了,训练时用的回调,显示的检测是验证集数据,蓝色是真实边界框,黄色是模型预测的边界框,有些教程其实搞错了,只有蓝色,没黄色其实是没检测到目标

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值