Python 图像转化为文档标签

#! /usr/bin/python

import os, sys

#正负样本路径

img_dirN="/home/lrj/picture/TestCutWord/NegetiveSam3000/"
img_dirP="/home/lrj/picture/TestCutWord/PositiveSam3000/"
fileListN=os.listdir(img_dirN)

fileListP=os.listdir(img_dirP)

#新建一个train.txt文件   默认在程序所在的文件夹中

train_txt=open('train.txt','w')  

for img in fileListP:
    str1=img+' '+'1'+'\n'        
    train_txt.writelines(str1)  
          

for img in fileListN:  
    str2=img+' '+'0'+'\n'  
    train_txt.writelines(str2)  
train_txt.close()