#! /usr/bin/python
import os, sys
img_dirP="/home/lrj/picture/TestCutWord/PositiveSam3000/"
fileListN=os.listdir(img_dirN)
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()
import os, sys
#正负样本路径
img_dirP="/home/lrj/picture/TestCutWord/PositiveSam3000/"
fileListN=os.listdir(img_dirN)
fileListP=os.listdir(img_dirP)
#新建一个train.txt文件 默认在程序所在的文件夹中
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()