1.GroupSoftmax:利用COCO和CCTSDB训练83类检测器
GroupSoftmax:利用COCO和CCTSDB训练83类检测器
在CV领域,工程师常利用YOLO、Faster RCNN、CenterNet等检测算法处理业务数据,旨在优化模型性能。然而,最新金花牛牛源码当模型在实际业务中发挥作用时,CEO的质疑往往紧随而来。为解决这一问题,我们设计了GroupSoftmax交叉熵损失函数,以解决模型训练的三大挑战。该函数允许类别合并,形成新的广东苹果源码组合类别,从而在训练时计算出各类别对应梯度,完成网络权重更新。理论上,GroupSoftmax交叉熵损失函数兼容多种数据集联合训练。
我们利用了COCO和CCTSDB数据集,基于Faster RCNN算法(SyncBN),源码制作开发联合训练了一个包含类的检测器。在COCO_minival测试集上,使用GroupSoftmax交叉熵损失函数训练的模型在mAP指标上提升了0.7个点,达到.3,相比原始Softmax交叉熵损失函数,性能显著提升。临沂系统源码此外,我们还训练了一个trident*模型,6个epoch在COCO_minival测试集上的mAP为.0,充分验证了GroupSoftmax交叉熵损失函数的有效性。
基于SimpleDet检测框架,我们实现了mxnet版本的源码名片报价GroupSoftmax交叉熵损失函数,并在GitHub上开源了源码。GroupSoftmax交叉熵损失函数的原理在于允许类别合并形成群组,计算群组类别概率的交叉熵损失,进而对激活值进行梯度计算。具体而言,当目标类别属于某个群组类别时,其梯度为群组类别梯度与子类别预测概率的比值。这样,GroupSoftmax交叉熵损失函数在处理类别合并情况时,能够有效更新网络权重。
实现GroupSoftmax交叉熵损失函数时,需要注意以下几点:
1. 对于未标注类别的数据集,可理解为与背景组成新的群组类别。
2. 在两阶段检测算法中,RPN网络应根据数据集特性调整为多分类,以适应模型训练需求。
3. 联合训练COCO和CCTSDB数据集时,最终分类任务为1+类,未标注类别的数据集可与背景组成组合类别。
4. 编写CUDA代码时,计算群组类别概率时,需加微小量避免分母为0导致的计算错误。
2024-11-19 07:571258人浏览
2024-11-19 07:512226人浏览
2024-11-19 06:292740人浏览
2024-11-19 06:221303人浏览
2024-11-19 06:061914人浏览
2024-11-19 05:481487人浏览
1.源代码泄露危害与保护措施2.如何防止源代码泄密?十种有效方法防止源代码泄密3.请问什么是源代码安全测试工具4.确保python代码的安全性5.软件公司要如何保障源代码的安全不会被外泄,不会被员工泄
1.易语言 打开网址源码2.用易语言做网页采集的时候都要把源代码转成什么编码?是不是固定的?如果不是,那有什么规律?易语言 打开网址源码 其实真的很简单,只要你打好易语言基础很容易做出来! .