yoloのアノテーション.txtをコピーしたい

#間違えて新規にアノテーションしちゃった！
同じ画像に対していくつもtxt作っちゃうんですよね。
そういうことあると思います。

アノテーションをコピーして追加or新規作成します。

##how to
フォルダを指定して、内部にあるテキストに対して無差別に動作します。
train.txtとかclass.txtとかあったら避難させてください。

番号ごとにコピーしたあと、同じ名前の txt を探して行を追加します。

arm_annoフォルダにあるアノテーションの2番をbody_annoにある同じ画像のアノテーションtxtに追加したければ、

$ python3 yolocopy.py arm_anno body_anno 2

となります。

##code

yolocopy.py

import glob
import os
import sys

path = os.getcwd()

_, infile, outfile, inline = sys.argv # target_file, output_file, target_number

for nowtxt in glob.glob(infile + "/*.txt"):
	textline = []
	with open(nowtxt) as f:
		l_strip = [s.strip() for s in f.readlines()]
		for k in l_strip:
			if str(k[0]) == str(inline):
				textline.append(k)
	basename = os.path.basename(nowtxt) #get name
	with open(outfile +'/'+ basename, mode='a') as f:
		for k in textline:
			f.write('\n' + str(k))
	
	trueline = []
	with open(outfile +'/'+ basename) as f:
		l_strip = [s.strip() for s in f.readlines()]
		for k in l_strip:
			if len(str(k))>0:
				trueline.append(k)
	#os.remove(outfile +'/'+ basename)
	with open(outfile +'/'+ basename, mode='w') as f:
		f.write('\n'.join(trueline))

##今後の改良点
同名の画像が存在するtxtに絞ればもっと手軽だと思うんですよね、面倒なのでしませんが。