你好，请问这个支持多个类别识别吗？ 我拿自己多类别的数据集训练的模型，检测的时候发现他的预测值返回类别都是2.0。如何解决呀？

[lareina115]

No description provided.

[ZeroE04]

支持。如果检测多类别，把代码里面num_classes都改成你的类别数目；然后再画框的地方，把画框的颜色自己调整一下，使不同类别颜色不同，我现在只用了一种颜色的框，也就是demo里面红蓝色的框。当然，你打标签的时候就得按照多类别来打

[pengyang1225]

你是怎么改修的？ cpp 文件的需要修改吗？

[ZeroE04]

你是怎么改修的？ cpp 文件的需要修改吗？

不需要，修改python中代码

[zhaowendao30]

你好，请问你问题解决了吗？我这多类别预测也都是2

[zwb170010]

你好，请问你问题解决了吗？我这多类别预测也都是2

[zhaowendao30]

将他模型中的hm参数改为2，还有数据加载和解码中的num_class改为2就行

…

------------------ 原始邮件 ------------------ 发件人: "ZeroE04/R-CenterNet" ***@***.***>; 发送时间: 2021年8月25日(星期三) 上午9:54 ***@***.***>; ***@***.******@***.***>; 主题: Re: [ZeroE04/R-CenterNet] 你好，请问这个支持多个类别识别吗？ 我拿自己多类别的数据集训练的模型，检测的时候发现他的预测值返回类别都是2.0。如何解决呀？ (#3) 你好，请问你问题解决了吗？我这多类别预测也都是2 — You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android.

[singlestrong]

将他模型中的hm参数改为2，还有数据加载和解码中的num_class改为2就行
…
------------------ 原始邮件 ------------------ 发件人: "ZeroE04/R-CenterNet" @.>; 发送时间: 2021年8月25日(星期三) 上午9:54 @.>; @.@.>; 主题: Re: [ZeroE04/R-CenterNet] 你好，请问这个支持多个类别识别吗？ 我拿自己多类别的数据集训练的模型，检测的时候发现他的预测值返回类别都是2.0。如何解决呀？ (#3) 你好，请问你问题解决了吗？我这多类别预测也都是2 — You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android.

请问多目标训练后，map是怎么得到的，怎么修改evaluation.py文件呐

[JiliuLiSU]

多类别预测是2的原因是predict.py代码有问题，改hm参数、num_class都没有用

[7777fsq]

多类别预测是2的原因是predict.py代码有问题，改hm参数、num_class都没有用

那请问应该怎么调整呢~~

哦 在predict.py里也有两处num_classes 也需要修改