glenn-jocher commited on
Commit
8a66eba
·
unverified ·
1 Parent(s): 601dbb8

Add `*.ts` to `VID_FORMATS` (#6859)

Browse files
Files changed (1) hide show
  1. 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 = ['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
 
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():