import os
import numpy as np
from PIL import Image
def get_img_name():
imglist = os.listdir('D:\\Downlode\\imagenet\\train')
imgnamefile = 'D:\\Downlode\\imagenet\\meta\\train.txt'
if os.path.exists(imgnamefile):
imgnamefile = open(imgnamefile, 'w')
# print(imgnamefile)
# imgnamefile.write("filename,label\n")
for i in imglist:
# img_name = i.split('.')[0] + '.' + i.split('.')[1] + ',' + i.split('_')[0]
img_name = i.split('.')[0] + '.' + i.split('.')[1]
label=i.split(('.'))[0]
if(label=="left"):
imgnamefile.write(img_name + ".jpg " + "0" + "\n")
if(label=="right"):
imgnamefile.write(img_name + ".jpg " + "1" + "\n")
# print(111)
# imgnamefile.writelines(img_name)
imgnamefile.close()
if __name__=='__main__':
get_img_name()