我正在尝试在定制的COCO数据集上训练Faster-RCNN模型。该数据集包含各种类别，总共16个类别（不同类型的鱼）。我的目标是从此数据集中仅隔离2个类别，并仅在这些类别上训练模型。
到目前为止，我已经尝试通过Github链接使用filter.py（https://github.com/immersive-limit/coco-manager），但这只能创建一个新的COCO注解。json
有什么办法可以过滤出类别和已注释的图像，以仅使用这些类别创建一个较小的新数据集？
非常感谢您的帮助！
谢谢。