search

airockchip / rknn_model_zoo

Apache License 2.0
1.01k stars 185 forks source link

yolov5_v7版本转rknn时,加载pt失败 #13

Closed wantao1008hh closed 1 year ago

wantao1008hh commented 1 year ago

pt文件应该没有破坏，使用yolov5中的detect.py可以正确检测到目标。路径应该也正确，用其他路径直接会说寻找不到pt文件。 sendpix0

wantao1008hh commented 1 year ago

版本问题，使用的是rknn-toolkit2最新的，不能用rknn-tookkit1.7.4版本，因为要是适配rk3568

wantao1008hh commented 1 year ago

torch.jit.load加载的是pt文件并不是yolov5默认训练好的best.pt，是torchscript模型，需要转换后，以pt重命名即可