Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add class filtering to LoadImagesAndLabels() dataloader #5172

Merged
merged 2 commits into from
Oct 13, 2021
Merged

Add class filtering to LoadImagesAndLabels() dataloader #5172

merged 2 commits into from
Oct 13, 2021

Commits on Oct 13, 2021

  1. 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)
```
    @glenn-jocher
    glenn-jocher authored Oct 13, 2021
    Configuration menu
    Copy the full SHA
    9638bdd View commit details
    Browse the repository at this point in the history

  2. segments fix

    @glenn-jocher
    glenn-jocher authored Oct 13, 2021
    Configuration menu
    Copy the full SHA
    fafab84 View commit details
    Browse the repository at this point in the history