open-mmlab / mmdeploy

OpenMMLab Model Deployment Framework
https://mmdeploy.readthedocs.io/en/latest/
Apache License 2.0
2.74k stars 628 forks source link

[Docs] deploy yolov3, yolox using rknn-toolkit and rknn-toolkit2 #1874

Open LEIJUNMIUI opened 1 year ago

LEIJUNMIUI commented 1 year ago

📚 The doc issue

yolov3, yolox for rknn-toolkit and rknn-toolkit2

partition_config = dict( type='rknn', # the partition policy name apply_marks=True, # should always be set to True partition_cfg=[ dict( save_file='model.onnx', # name to save the partitioned onnx start=['detector_forward:input'], # [mark_name:input, ...] end=['yolo_head:input'], # [mark_name:output, ...] output_names=[f'pred_maps.{i}' for i in range(3)]) # output names ])

Suggest a potential alternative/fix

这个文档在转换yolov3过程中直接复制进去吗 还是需要根据实际情况修改参数 如何修改

AllentDan commented 1 year ago

📚 The doc issue

yolov3, yolox for rknn-toolkit and rknn-toolkit2

partition_config = dict( type='rknn', # the partition policy name apply_marks=True, # should always be set to True partition_cfg=[ dict( save_file='model.onnx', # name to save the partitioned onnx start=['detector_forward:input'], # [mark_name:input, ...] end=['yolo_head:input'], # [mark_name:output, ...] output_names=[f'pred_maps.{i}' for i in range(3)]) # output names ])

Suggest a potential alternative/fix

这个文档在转换yolov3过程中直接复制进去吗 还是需要根据实际情况修改参数 如何修改

直接复制进去就行,除非你在网络 neck 部分将三层语义层输出改成其两层输出或者一层输出