Open LEIJUNMIUI opened 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 部分将三层语义层输出改成其两层输出或者一层输出
📚 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过程中直接复制进去吗 还是需要根据实际情况修改参数 如何修改