make coco tex

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()

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值