mrwangwg123 / my-rknn-yolov5 Goto Github PK

|-- README.md
|-- commons                         # 帮助工具
|   |-- rknn-convert                # 转换其他框架的模型到rknn,包括(tensorflow、tflite、caffe、onnx)
|   `-- yolov5-torch2rknn-convert   # 将torch框架训练的yolov5的模型转成rknn支持的onnx
|       |-- common.py
|       |-- common_rk_plug_in.py    # rknn 中兼容的网络层
|       |-- experimental.py
|       |-- export.py               # 将torch框架训练的yolov5的模型转成rknn支持的onnx,内部做了兼容
|       |-- yolo.py                 # 自动生成torch下的yolov5网络
|       |-- yolov5l.yaml 
|       |-- yolov5m.yaml
|       |-- yolov5s.yaml
|       `-- yolov5x.yaml
|-- cpp                             # 使用rknn部署的yolov5算法的C++示例
|-- data                            # 数据/测试图片
|-- docs
|-- model                           # 模型文件（待转换的模型、待测试的模型）
|   |-- face-focus-removed.onnx     # python代码中使用的人脸检测测试模型（转换后的）
|   |-- face-model.pt               # 人脸检测训练好的模型，需要使用export.py转换成rknn
|   `-- yolov5s.onnx                # yolov5示例转换好的模型，
|-- packages                        # rknn-toolkit包
|-- python                          # 使用rknn部署的yolov5算法的python示例
|   |-- detector_bridge.py
|   |-- rknn_impl.py
|   |-- test.py                     # 示例程序入口
|   |-- torch_impl.py
|   `-- yolov5_detector_impl.py
`-- runs                            # 运行结果暂存

(rknn-toolkit) autobuild@ubuntu-server-117:~/wangwengang/my-rknn-yolov5/commons/yolov5-torch2rknn-convert$ python models/export.py --weights /home/autobuild/wangwengang/my-rknn-yolov5/model/face-model.pt --img 640 --rknn_mode
Namespace(batch_size=1, device='cpu', dynamic=False, grid=False, img_size=[640, 640], rknn_mode=False, weights='/home/autobuild/wangwengang/my-rknn-yolov5/model/face-model.pt')
Using torch 1.5.1 CPU

weight :  /home/autobuild/wangwengang/my-rknn-yolov5/model/face-model.pt
Traceback (most recent call last):
  File "models/export.py", line 39, in <module>
    model = attempt_load(opt.weights, map_location=device)  # load FP32 model
  File "/home/autobuild/wangwengang/my-rknn-yolov5/commons/yolov5-torch2rknn-convert/models/experimental.py", line 119, in attempt_load
    ckpt = torch.load(w, map_location=map_location)  # load
  File "/home/autobuild/miniconda3/envs/rknn-toolkit/lib/python3.6/site-packages/torch/serialization.py", line 592, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/autobuild/miniconda3/envs/rknn-toolkit/lib/python3.6/site-packages/torch/serialization.py", line 852, in _load
    result = unpickler.load()
AttributeError: Can't get attribute 'SiLU' on <module 'torch.nn.modules.activation' from '/home/autobuild/miniconda3/envs/rknn-toolkit/lib/python3.6/site-packages/torch/nn/modules/activation.py'>