rockchip_rv1109_rknn-yolov3-demo icon indicating copy to clipboard operation
rockchip_rv1109_rknn-yolov3-demo copied to clipboard

你好,我想请问,如果要换成自己训练的yolov3模型?需要修改代码哪些部分呢

Open tiger-98 opened this issue 3 years ago • 7 comments

tiger-98 avatar Jan 13 '22 08:01 tiger-98

修改src/里两个文件就可以了,注意自己训练的yolov3模型的参数匹配和2rknn转换细节处理。

BaronLeeLZP avatar Jan 15 '22 00:01 BaronLeeLZP

你好,请问我这边基于yolov5和.pt权重文件,是否直接转换成rknn替换您开源的这个,就可以了呢?

mysteryljn avatar Mar 12 '22 15:03 mysteryljn

不可以的,v5的后处理和v3的不一样,你可以参考瑞芯微提供的yolov5的demo,将自己转换的模型放到那里去

tiger-98 avatar Mar 13 '22 01:03 tiger-98

您好,能给指条路嘛,来个链接

mysteryljn avatar Mar 18 '22 13:03 mysteryljn

https://github.com/airockchip/rknn_model_zoo/tree/main/models/vision/object_detection/yolov5-pytorch

tiger-98 avatar Apr 01 '22 09:04 tiger-98

@BaronLeeLZP 请问3588可以直接用这个后处理代码吗?不会c++只能当伸手党了

Answergeng avatar Sep 26 '23 02:09 Answergeng

您好,这里是qq邮箱收到自动回复,您的信件已经收到,祝好!

tiger-98 avatar Sep 26 '23 02:09 tiger-98