将E:\yolov5-5.0\data\mydataset\images\train该路径下所有图片路径写入E:\yolov5-5.0\data\mydataset\train.txt该如何实现
时间: 2023-09-04 15:12:58 浏览: 70
你可以使用Python编写一个程序来实现将指定路径下所有图片路径写入指定的文本文件。具体代码如下:
```python
import os
# 遍历指定目录下所有文件及子目录,并把所有图片路径写入文本文件
def write_images_path_to_txt(images_dir, output_file):
with open(output_file, 'w') as f:
for root, dirs, files in os.walk(images_dir):
for file in files:
if file.endswith('.jpg') or file.endswith('.png'):
image_path = os.path.join(root, file)
f.write(image_path + '\n')
# 调用函数将图片路径写入文本文件
images_dir = r'E:\yolov5-5.0\data\mydataset\images\train'
output_file = r'E:\yolov5-5.0\data\mydataset\train.txt'
write_images_path_to_txt(images_dir, output_file)
```
在运行程序之前,请先确认指定路径下确实存在要写入的图片,程序会遍历指定目录下所有文件及子目录,并把所有图片路径写入指定的文本文件。
阅读全文