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

幸狐开发板官网：

https://wiki.luckfox.com/zh/Luckfox-Pico-Ultra/RKNN

参考这部分

git clone git@gitcode.com:qq_36813351/retinaface-pytorch.git

git clone git@gitcode.com:qq_36813351/facenet-pytorch.git

git clone git@gitcode.com:qq_36813351/airockchipyolov5.git

retinaface-pytorch和facenet-pytorch按照官方文档运行就没问题。

airockchipyolov5这个有一个问题：

错误修改：

git clone git@gitcode.com:qq_36813351/airockchipyolov5.git

这个按照官网的手顺去做他有问题： python export.py --rknpu --weight yolov5s.pt 他就报错。

背景： PyTorch 2.6 版本将 torch.load 函数的默认参数从 weights_only=False 改为了 True。这意味着现在默认只允许加载“纯权重”数据。​ 冲突： 你使用的 YOLOv5 代码库（特别是 models/yolo.py 中的 Model 类）定义了自定义的模型结构。当 PyTorch 尝试读取 .pt 文件时，发现里面引用了这个自定义类，因此触发了安全警告并中断加载。​ 信任问题： PyTorch 要求你必须明确表示“我信任这个文件”，否则拒绝加载。​

所以就改成下图这样：​

yolov5/models/experimental.py​

找到第 80 行左右的代码：​

ckpt = torch.load(attempt_download(w), map_location='cpu')​

ckpt = torch.load(attempt_download(w), map_location='cpu', weights_only=False)

改完模型转化就没问题了。