• Glenn Jocher's avatar
    Add class filtering to `LoadImagesAndLabels()` dataloader (#5172) · a3469269
    Glenn Jocher 提交于
    * 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
    a3469269
datasets.py 43.6 KB