Jetson Nano 【5】Pytorch-YOLOv3原生模型测试

  • 在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加速,看看能到什么程度
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

椰子奶糖

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值