python删除coco数据库中的小人

最新推荐文章于 2023-05-16 17:50:13 发布

原创最新推荐文章于 2023-05-16 17:50:13 发布 · 260 阅读

0 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

34 篇文章

订阅专栏

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

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