保存文件夹中的文件路径(绝对路径)。

# !/usr/bin/env python
# -*- encoding: utf-8 -*-

import os

img_path = '/home/jjuv/Downloads/yolo_data/' #图片的绝对路径,注意最后加上 /
save_txt = './train.txt' #保存在txt文件中。
img_list = os.listdir(img_path)
print('img_list: ', img_list)

with open(save_txt, 'w') as f:
    for img_name in img_list:
        f.write(img_path+img_name + '\n')
Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐