文章浏览阅读122次。该代码段展示了如何利用Python的os模块获取指定路径下特定类型图片的绝对路径，并将这些路径写入txt文件中，适用于Yolo项目的数据准备。用户需输入图片路径、txt文件名和图片文件类型，程序会遍历目录并将符合类型的文件路径写入文本。
如标题
可以用于yolo项目中所需要的写入图片的绝对路径那一个环节
也可以用于其他的情况（博主想不出来）
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
import os
file_name_1=input('输入要创建准备写入txt的的文件名称：')
file_1=open(f'{file_name_1}.txt',mode='w',encoding='utf-8')
path_1=input('输入图片的路径： ')
type_1=input('输入写入txt的文件后缀名称：')
sum1s=os.listdir(path_1)
# 把文件输入的路径下的文件遍历
for sum1 in sum1s:
sum2=os.path.splitext(sum1)
#判断后缀名称是不是输入类型的（sum2[1]为输入类型的后缀名）
if sum2[1]==f'.{type_1}':
sum3=os.getcwd()
sum5=f'{sum3}{sum1}'
file_1.writelines(f'{sum5}\n')
print(f'已写入 {sum5}到{file_name_1}.txt中')
|