首页 PaddleCV 帖子详情
PaddleDetection 如何训练指定label
收藏
快速回复
PaddleCV 问答目标检测 1153 11
PaddleDetection 如何训练指定label
收藏
快速回复
PaddleCV 问答目标检测 1153 11

1:下载的自定义数据集有8个标签,如何训练时只训练其中3个或者2个

2:deploy/infer  识别时候只识别部分label如何处理。(修改infer.xml 会报错getColorList 数组越界)

0
收藏
回复
全部评论(11)
时间顺序
FutureSI
#2 回复于2021-09

做下数据预处理,把需要的数据集分离出来

0
回复
FutureSI
#3 回复于2021-09

应该说是把需要的标注分离出来

0
回复
◆丶_____秋
#4 回复于2021-09
做下数据预处理,把需要的数据集分离出来

好的谢谢  我按照这个思路试试

0
回复
JavaRoom
#5 回复于2021-09

不错不错

0
回复
深渊上的坑
#6 回复于2021-09

如果只是deploy/infer预测的时候,可以修改下源代码写个逻辑,如果匹配到某个特定标签,如何如何,如果是不需要的,略过

0
回复
深渊上的坑
#7 回复于2021-09

这里有个类似的例子供参考,比如在树莓派部署安全帽检测模型的时候:https://aistudio.baidu.com/aistudio/projectdetail/1209733

做了一个判断逻辑(C++)

for (int i = 0; i < results.size(); i++) {
        printf("[%d] %s - %f %f,%f,%f,%f\n", i, results[i].class_name.c_str(),
            results[i].score, results[i].left, results[i].top, results[i].right,
            results[i].bottom);
        // 增加一个判断,如果检测到目标,发出报警、保存截图
        if (results[i].class_name == "head") {
            // 触发报警
            warn();
            std::time_t t = std::time(0);
            // 这里是用时间戳存储保存的图片,保证有序,还可以继续优化
            sprintf(saveName, "/home/pi/Paddle-Lite-Demo/PaddleLite-armlinux-demo/object_detection_demo/output/%ld.jpg", t);
            cv::imwrite(saveName, output_image);
        }
    }
0
回复
深渊上的坑
#8 回复于2021-09

如果是在训练的时候,只指定其中几个,感觉要对标注文件进行修改了

0
回复
深渊上的坑
#9 回复于2021-09

可以参考这个项目

飞桨2.0 PaddleDetection:瓶装酒瑕疵检测迁移学习教程

https://aistudio.baidu.com/aistudio/projectdetail/250844

数据清洗章节,这里处理了背景类的标注

其实就是对json文件进行解析,然后修改

0
回复
深渊上的坑
#10 回复于2021-09

如果是VOC格式,要么就是一个个标注文件遍历过去改;要么就是先转成json格式再改

0
回复
◆丶_____秋
#11 回复于2021-10
如果是VOC格式,要么就是一个个标注文件遍历过去改;要么就是先转成json格式再改

回来看到了回复   提供了很多思路  感谢

0
回复
李长安
#12 回复于2021-11
这里有个类似的例子供参考,比如在树莓派部署安全帽检测模型的时候:https://aistudio.baidu.com/aistudio/projectdetail/1209733 做了一个判断逻辑(C++) [代码]
展开

坑姐厉害呀!!!!!会写C++

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户