RNNK Demo代码(retinaface,facenet,airockchipyolov5)模型转化遇到的问题

张开发
• 2026/4/17 1:03:05 • 15 分钟阅读

最新文章

推荐文章

相关文章

分享文章

RNNK Demo代码(retinaface,facenet,airockchipyolov5)模型转化遇到的问题
幸狐开发板官网https://wiki.luckfox.com/zh/Luckfox-Pico-Ultra/RKNN参考这部分git clone gitgitcode.com:qq_36813351/retinaface-pytorch.gitgit clone gitgitcode.com:qq_36813351/facenet-pytorch.gitgit clone gitgitcode.com:qq_36813351/airockchipyolov5.gitretinaface-pytorch和facenet-pytorch按照官方文档运行就没问题。airockchipyolov5这个有一个问题错误修改git clone gitgitcode.com:qq_36813351/airockchipyolov5.git这个按照官网的手顺去做他有问题 python export.py --rknpu --weight yolov5s.pt 他就报错。背景 PyTorch 2.6 版本将 torch.load 函数的默认参数从 weights_onlyFalse 改为了 True。这意味着现在默认只允许加载“纯权重”数据。​ 冲突 你使用的 YOLOv5 代码库特别是 models/yolo.py 中的 Model 类定义了自定义的模型结构。当 PyTorch 尝试读取 .pt 文件时发现里面引用了这个自定义类因此触发了安全警告并中断加载。​ 信任问题 PyTorch 要求你必须明确表示“我信任这个文件”否则拒绝加载。​所以就改成下图这样​yolov5/models/experimental.py​找到第 80 行左右的代码​ckpt torch.load(attempt_download(w), map_locationcpu)​ckpt torch.load(attempt_download(w), map_locationcpu, weights_onlyFalse)改完模型转化就没问题了。

更多文章