# include image suffixes
IMG_FORMATS = 'bmp', 'dng', 'jpeg', 'jpg', 'mpo', 'png', 'tif', 'tiff', 'webp', 'pfm'
# include video suffixes
VID_FORMATS = 'asf', 'avi', 'gif', 'm4v', 'mkv', 'mov', 'mp4', 'mpeg', 'mpg', 'ts', 'wmv'
def is_img(file_path: str):
return file_path.split('.')[-1] in IMG_FORMATS
def is_video(file_path: str):
return file_path.split('.')[-1] in VID_FORMATS
判断文件是不是图片或者视频
最新推荐文章于 2024-06-14 15:32:43 发布
该代码定义了两个函数,is_img和is_video,分别用于判断文件路径是否对应于图像或视频文件。它们基于文件扩展名来识别不同的格式,如bmp、jpeg、png等是图像格式,而asf、avi、mp4等是视频格式。
摘要由CSDN通过智能技术生成