toschi23
GitHub
2 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
utils/datasets.py
View File
| @@ -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 | |||
Loading…