我正在尝试训练一个可以检测旋转矩形的模型。我找到了Nvidia ODTK，但我没有发现任何迹象表明我是否可以只使用一个类(类别)来训练它。默认的分类数是80，现在我不确定只有一个分类是否可以很好地工作。
尝试使用--classes参数。看到https://github.com/NVIDIA/retinanet-examples/blob/main/TRAINING.md其他参数