Add `*.ts` to `VID_FORMATS` (#6859)
Browse files- utils/datasets.py +2 -2
utils/datasets.py
CHANGED
@@ -33,8 +33,8 @@ from utils.torch_utils import torch_distributed_zero_first
|
|
33 |
|
34 |
# Parameters
|
35 |
HELP_URL = 'https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data'
|
36 |
-
IMG_FORMATS =
|
37 |
-
VID_FORMATS =
|
38 |
|
39 |
# Get orientation exif tag
|
40 |
for orientation in ExifTags.TAGS.keys():
|
|
|
33 |
|
34 |
# Parameters
|
35 |
HELP_URL = 'https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data'
|
36 |
+
IMG_FORMATS = 'bmp', 'dng', 'jpeg', 'jpg', 'mpo', 'png', 'tif', 'tiff', 'webp' # include image suffixes
|
37 |
+
VID_FORMATS = 'asf', 'avi', 'gif', 'm4v', 'mkv', 'mov', 'mp4', 'mpeg', 'mpg', 'ts', 'wmv' # include video suffixes
|
38 |
|
39 |
# Get orientation exif tag
|
40 |
for orientation in ExifTags.TAGS.keys():
|