- 在torch1.4 版本下训练出的一个模型,正好用于测试Nano的环境安装是否正确
-
首先准备好测试程序、模型、测试数据
- 这些东东在本地是跑过的,是可以跑,就是速度堪忧,此次测试的目的主要是看环境能否跑,而来看看原生(不是tensorRT的情况下)的速度
-
所谓原生,是基于:原生git地址这个仓库未TensorRT的模型
- 先贴下基本结构
- 我们用3号程序choose得到samples后,就可以开始4号程序测试(直接运行即可)
- 以下是测试结果:
- 下面就是目标的坐标,然后我们进outputs看看
- 得到了一些新图片:
- 抽了几张可以看到,成功运行,但是在密集小目标的场景下表现不佳,比如此处,后续需要进一步改进
遇到的错误
- FileNotFoundError: [Errno 2] No such file or directory:
- 这是一个典型的路径错误,经排查,是由于linux与windows的路径斜杠不同,我们只需要改成linux中的"/"即可解决
- too many indices for tensor of dimension 1
- 这个错误是说太多的一维张量,起初没搞懂啥意思,于是在报错处打了个断点开始debug
- 发现这句是调用上句的detections,于是看了detections的值
- 发现它并没有什么值,也就是说,大概率是这一句出了错,从而导致下面运行不下去
- 解决:发现是自己的失误,篡改了源码,方法调用错误,是在初期测试的时候改的,忘记填的坑
- 正确方法如下
总结
- 此次测试是基本成功的,原生是可以跑的,还顺便填了个坑,
- 至于速度,大概是1-3fps,意料之中(还是比本地快一丢丢)
- 下次准备研究研究官方的demo,使用tensorRT加速,看看能到什么程度