【接上文】mediapipe进行目标检测

上文提到需要自定义一个模型,参考了探索MediaPipe自定义机器学习模型_mediapipe训练自己的数据集-CSDN博客,但是实际运行一直无法加载Google.colab的库,至于为啥,我也不知道。。。。。也没有找到比较好的解决方案,只能换途径。

再次理解这个项目的整个逻辑顺序,mediapipe进行目标检测,但是自带模型检测的种类有限,需要自定义模型。到此,拟方案是,采用tensorflow框架去训练出模型,逻辑顺序是

1、准备数据集,

采用labelimg对原始图片进行标注,得到xml文件。这里注意labelimg适配python3.8

将xml文件转化为csv文件。

2、加载基础模型,采用的是medaipipe的原始模型efficientdet_lite0

3、训练模型

这里需要导入的tflite_model_maker 的库,pip install 包 -i 阿里源,建议阿里源速度快一点,但是仍然一直中断,报错,研究发现可能是tflit_model_maker里面的某个包,适配大于3.9的python版本,于是又改用python3.10,但是网络不稳定,下载着实相当麻烦

4、导出模型

5、将自定义的模型再倒回上文的检测.py,进行验证,测试

以上是整个逻辑。

至此把逻辑弄通,但是还未实践成功,若成功,下文贴代码。

  • 12
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值