search

rockchip-linux / rknpu

BSD 3-Clause "New" or "Revised" License
204 stars 60 forks source link

YOLOv5 #11

Open LittleRain626 opened 3 years ago

LittleRain626 commented 3 years ago

如果使用rknn-toolkit的example文件夹下的代码去转换YOLOv5的Pytorch模型应该具体怎么操作呢？

LittleRain626 commented 3 years ago

请问作者具体是如何修改YOLOv5的pytorch模型的，万分感谢！

kk52099 commented 3 years ago

https://github.com/rockchip-linux/rknpu/issues/11#issuecomment-857456675 同问，修改了哪里很重要， 我们才能用这个网络训练