Glenn Jocher a346926996 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

2021-10-13 15:48:54 -07:00

..

aws

Fix yaml.safe_load() ignore emoji errors (#5060)

2021-10-05 13:41:52 -07:00

flask_rest_api

Standardize headers and docstrings (#4417)

2021-08-14 21:17:51 +02:00

google_app_engine

Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561)

2021-06-09 21:14:09 +02:00

loggers

Update W&B README (#5006)

2021-09-30 00:35:00 -07:00

__init__.py

Update sys.path.append(str(ROOT)) (#4852)

2021-09-18 15:02:08 +02:00

activations.py

Standardize headers and docstrings (#4417)

2021-08-14 21:17:51 +02:00

augmentations.py

Update default Albumentations (#4931)

2021-09-26 12:10:33 -07:00

autoanchor.py

Refactor Detect() anchors for ONNX <> OpenCV DNN compatibility (#4833)

2021-10-11 09:58:42 -07:00

callbacks.py

Optimised Callback Class to Reduce Code and Fix Errors (#4688)

2021-09-07 18:32:15 +02:00

datasets.py

Add class filtering to LoadImagesAndLabels() dataloader (#5172)

2021-10-13 15:48:54 -07:00

downloads.py

Replace os.system('unzip file.zip') -> ZipFile.extractall() (#4919)

2021-09-25 08:52:36 -07:00

general.py

Fix pylint: do not use bare 'except' (#5025)

2021-10-03 17:54:40 -07:00

loss.py

Remove redundant ComputeLoss code (#4701)

2021-09-08 12:48:33 +02:00

metrics.py

Close matplotlib plots after opening (#4612)

2021-08-30 17:22:21 +02:00

plots.py

Update plots.py (#5127)

2021-10-11 01:59:22 -07:00

torch_utils.py

Consolidate init_seeds() (#4849)

2021-09-18 13:28:42 +02:00