提交 5286eae2 编写于 作者: M MaoXianxin

用OpenCV实现超轻量的NanoDet目标检测模型!

上级 a05f4ccf
......@@ -74,10 +74,12 @@ nanodet通过一些论文里的trick组合起来得到了一个兼顾精度、
在编写完调用opencv的做nanodet目标检测的程序后,为了验证程序的有效性,从COCO数据集里选取几张图片测试并且与官方代码做比较,官方代码是用python编写的调用pytorch框架做目标检测的。结果如下,左边的图是官方代码的可视化结果,右边的图是opencv做nanodet目标检测的可视化结果。
![](https://maoxianxin1996.oss-accelerate.aliyuncs.com/codechina/20210604185122.png)![Image](https://mmbiz.qpic.cn/sz_mmbiz_jpg/gYUsOT36vfoUYV6mWLdPe5kV2OI9VlvYmRdvI6aZkwQEvAicRiccFuDlZSIXKDD0uG3pcdw49NuJKFn0hyZqicI0w/640?wx_fmt=jpeg&tp=webp&wxfrom=5&wx_lazy=1&wx_co=1)
![](https://maoxianxin1996.oss-accelerate.aliyuncs.com/codechina/20210604190223.png)
把官方代码和我编写的代码做了一个性能比较的实验,实验环境是ubuntu系统,8G显存的gpu机器。在实验中读取一个视频文件,对视频里的每一帧做目标检测,分别运行官方的调用pytorch框架做目标检测的python代码和我编写的调用opencv做目标检测的python代码,在terminal终端输入top查看这两个程序运行时占用的内存,截图如下。第一行是opencv做nanodet目标检测程序运行时占用的内存,第二行是官方代码运行时占用的内存,可以看到使用opencv做nanodet目标检测对内存的消耗明显要低于官方代码的pytorch框架做nanodet目标检测的。
![](https://maoxianxin1996.oss-accelerate.aliyuncs.com/codechina/20210604185137.png)
GitHub传送门:[https://github.com/hpc203/nanodet-opncv-dnn-cpp-python](https://github.com/hpc203/nanodet-opncv-dnn-cpp-python)
\ No newline at end of file
GitHub传送门:[https://github.com/hpc203/nanodet-opncv-dnn-cpp-python](https://github.com/hpc203/nanodet-opncv-dnn-cpp-python)
![](https://maoxianxin1996.oss-accelerate.aliyuncs.com/codechina/20210603140942.png)
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册