os.path.splitext()的用法
python
·
在yolox的代码中看到代码如下:
def get_image_list(path):
image_names = []
for maindir, subdir, file_name_list in os.walk(path):
for filename in file_name_list:
apath = os.path.join(maindir, filename)
ext = os.path.splitext(apath)[1]
if ext in IMAGE_EXT:
image_names.append(apath)
return image_names
IMAGE_EXT = [".jpg", ".jpeg", ".webp", ".bmp", ".png"]
os.path.splitext(apath)[1]的作用是分离文件名与扩展名,结果以元组返回
>>> zz = 'J:/data/test_result_0424_preprocess/1_depth_resize.png'
>>> import os
>>> os.path.splitext(zz)[1]
'.png'
>>> os.path.splitext(zz)
('J:/data/test_result_0424_preprocess/1_depth_resize', '.png')
>>> os.path.splitext(zz)[0]
'J:/data/test_result_0424_preprocess/1_depth_resize'
这样在得到图像数据的时候就可以留下自己想要的格式,及过滤掉.db文件等

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。
更多推荐
所有评论(0)