vid_formats bug fix #146

5 years ago · 5c2cd711df
parent 0a09882e4f
commit 5c2cd711df
1 changed files with 1 additions and 1 deletions
--- a/utils/datasets.py
+++ b/utils/datasets.py
@ -18,7 +18,7 @@ from utils.utils import xyxy2xywh, xywh2xyxy

 help_url = 'https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data'
 img_formats = ['.bmp', '.jpg', '.jpeg', '.png', '.tif', '.dng']
-vid_formats = ['.mov', '.avi', '.mp4']
+vid_formats = ['.mov', '.avi', '.mp4', '.mpg']

 # Get orientation exif tag
 for orientation in ExifTags.TAGS.keys():