qinggangwu / yolov7-pose_Npoint_Ncla

基于yoloV7-pose添加任意个关键点和检测目标多分类
Other
112 stars 17 forks source link

大佬,有将模型导出为 .onnx 的脚本吗 #8

Closed yechens closed 1 year ago

yechens commented 1 year ago

官方的 export.py 脚本似乎不能兼容我们的代码

yechens commented 1 year ago

Problem solved

wzy0903 commented 1 year ago

你好,你解决了吗,怎么导出和用opencv推理呀

RunpuWei commented 1 year ago

你好,你解决了吗,怎么导出和用opencv推理呀

utils/general.py 代码内,将non_max_suppression_export()方法的nkpt参数修改为关键点的个数。