Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

Selecting custom classes from dataset #68

Open
Manjuphoenix opened this issue Jul 16, 2023 · 0 comments
Open

Selecting custom classes from dataset #68

Manjuphoenix opened this issue Jul 16, 2023 · 0 comments

Comments

@Manjuphoenix
Copy link

I found using "MODEL.ROI_HEADS.TRAIN_CLASSES = ['class_ids']" in config.py and setting TRAIN_CLASSES: ['class_ids'] inside the faster_rcnn_VGG.yaml might help but it does not work as expected.
Has anyone tried to filter particular classes by class_names or class_ids in the config rather than modifying the dataset itself?

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant