toschi23
commited on
Add `*.asf` video support (#6436)
Browse files- utils/datasets.py +1 -1
utils/datasets.py
CHANGED
@@ -34,7 +34,7 @@ from utils.torch_utils import torch_distributed_zero_first
|
|
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 = ['avi', 'gif', 'm4v', 'mkv', 'mov', 'mp4', 'mpeg', 'mpg', 'wmv'] # include video suffixes
|
38 |
DEVICE_COUNT = max(torch.cuda.device_count(), 1)
|
39 |
|
40 |
# Get orientation exif tag
|
|
|
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', 'wmv'] # include video suffixes
|
38 |
DEVICE_COUNT = max(torch.cuda.device_count(), 1)
|
39 |
|
40 |
# Get orientation exif tag
|