python删除coco数据库中的小人

刀么克瑟拉莫

于 2020-08-13 16:41:59 发布

阅读量243

点赞数 1

分类专栏： python

本文链接：https://blog.csdn.net/random_repick/article/details/107983516

版权

python 专栏收录该内容

34 篇文章

订阅专栏

本文介绍了一种Python脚本，用于处理特定目录下的文件。脚本遍历指定路径下的所有文件，对每个文件进行读取并筛选出符合条件的行，然后将这些行写入到新的文件中。特别地，脚本会跳过文件名最后一个字符不是't'的文件，并且只处理那些特定条件下的行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os

path='/home/lwd/data/obstacle/train-person/train'
files=os.listdir(path)
files.sort()
for f in files:
  if f[-1]!='t':
    continue
  o=open('/home/lwd/data/obstacle/train-person/big/'+f,'w')
  count=len(open(path+'/'+f,'rU').readlines())
  cnt=0
  for l in open(path+'/'+f):
    if l[0]!='0':
      o.write(l)
      cnt+=1
      if cnt!=count:
        o.write('\n')
      continue
    str_list=l.split()
    if float(str_list[-1])*float(str_list[-2]) > 0.05:
      o.write(l)
      cnt+=1
      if cnt!=count:
        o.write('\n')
  o.close()