是否可以使用仅一个类的 Nvidia ODTK 训练模型？

【问题标题】：Is it possible to Train an Model with Nvidia ODTK with only one class?是否可以使用仅一个类的 Nvidia ODTK 训练模型？
【发布时间】：2021-10-29 18:07:39
【问题描述】：

我正在尝试训练一个可以检测旋转矩形的模型。我找到了 Nvidia ODTK，但我没有发现任何迹象表明我是否可以只用一个类（类别）来训练它。默认的类别数是 80，现在我不确定它是否可以只使用一个。

【问题讨论】：

标签： object-detection nvidia custom-training

【解决方案1】：

尝试使用--classes 参数。见https://github.com/NVIDIA/retinanet-examples/blob/main/TRAINING.md#other-arguments

【讨论】：