Skip to content

Y
yolov5
Unverified 提交 c9bda112 authored 作者: Transigent's avatar Transigent 提交者： GitHub
浏览文件
操作

Update utils/datasets.py to support .webp files (#2174)

Simply added 'webp' as an image format to the img_formats array so that webp image files can be used as training data.
上级 ace3e02e
隐藏空白字符变更
内嵌 并排
正在显示 包含 1 行增加 和 1 行删除
utils/datasets.py
浏览文件 @ c9bda112
...@@ -25,7 +25,7 @@ from utils.torch_utils import torch_distributed_zero_first ...@@ -25,7 +25,7 @@ from utils.torch_utils import torch_distributed_zero_first
# Parameters # Parameters
help_url = 'https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data' help_url = 'https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data'
img_formats = ['bmp', 'jpg', 'jpeg', 'png', 'tif', 'tiff', 'dng'] # acceptable image suffixes img_formats = ['bmp', 'jpg', 'jpeg', 'png', 'tif', 'tiff', 'dng', 'webp'] # acceptable image suffixes
vid_formats = ['mov', 'avi', 'mp4', 'mpg', 'mpeg', 'm4v', 'wmv', 'mkv'] # acceptable video suffixes vid_formats = ['mov', 'avi', 'mp4', 'mpg', 'mpeg', 'm4v', 'wmv', 'mkv'] # acceptable video suffixes
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论