Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add class filtering to
LoadImagesAndLabels()
dataloader (#5172)
* Add train class filter feature to datasets.py Allows for training on a subset of total classes if `include_class` list is defined on datasets.py L448: ```python include_class = [] # filter labels to include only these classes (optional) ``` * segments fix
- Loading branch information