上文提到需要自定义一个模型,参考了探索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,进行验证,测试
以上是整个逻辑。
至此把逻辑弄通,但是还未实践成功,若成功,下文贴代码。