0

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

4

1 回答 1

0

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

于 2021-11-04T00:36:53.227 回答