实例:从文件夹中批量挑选文件
(挑选出后缀名为jpg的图片)
import os
test_dir ='./JPEGImages'
#获取测试文件夹内的文件列表list
pics = os.listdir(test_dir)
for im in pics:
if os.path.splitext(im)[1]=='.jpg':
#(或)if im.endswith('jpg'):
print(os.path.join(test_dir,im))
os module:
os.path.split() 分裂上层目录和下层目录|文件 ,生成一个包含两个元素的元组。
>>> os.path.split("/opt/workspace/tab.py")
('/opt/workspace', 'tab.py') #tab.py为文件
os.path.basename(path) 返回path最后的文件|目录名
>>> os.path.basename("/opt/workspace/tab.py") #本质 == os.path.split("/opt/workspace/tab.py")[1]
'tab.py'
os.path.join() 合并目录名和文件名|目录名
os.path.splitext() 分离文件名和扩展名
>>> os.path.splitext('000-0.jpg')
('000-0', '.jpg')
os.path.abspath() 获得绝对路径
>>> os.path.abspath('000-0.jpg')
'/home/ycc/darknet-master/yandata/VOC2008/000-0.jpg'
os.path.getsize() 获得文件或目录的大小
>>> os.path.getsize('000-0.jpg')
6871
os.listdir() 获取指定目录下的文件和目录名列表