【ABCNet】训练自己的模型(三)

【ABCNet训练自己的模型(一)】
【ABCNet训练自己的模型(二)】

预测

上一节里面说了怎么修改参数训练我们自己的数据，得到自己的模型，这一节说一下预测怎么修改参数对最后一层的结果进行解码

修改解码标签和类别数量

设置推理过程中置信度阀值

这个函数就在demo.py里面

其他有什么问题可以留言。