From 3b7ac28ed1c760ee3ed6a9780027a4a3e775f937 Mon Sep 17 00:00:00 2001 From: toschi23 Date: Wed, 26 Jan 2022 14:26:02 +0100 Subject: [PATCH] Add `*.asf` video support (#6436) --- utils/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index fa73cba..4e0c38e 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -34,7 +34,7 @@ from utils.torch_utils import torch_distributed_zero_first # Parameters HELP_URL = 'https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data' IMG_FORMATS = ['bmp', 'dng', 'jpeg', 'jpg', 'mpo', 'png', 'tif', 'tiff', 'webp'] # include image suffixes -VID_FORMATS = ['avi', 'gif', 'm4v', 'mkv', 'mov', 'mp4', 'mpeg', 'mpg', 'wmv'] # include video suffixes +VID_FORMATS = ['asf', 'avi', 'gif', 'm4v', 'mkv', 'mov', 'mp4', 'mpeg', 'mpg', 'wmv'] # include video suffixes DEVICE_COUNT = max(torch.cuda.device_count(), 1) # Get orientation exif tag