import os
import os.path as osp
import shlex
import shutil
import subprocess
model = 'tiny-yolo-voc-person'
max_iterations = 0
for model_fname in os.listdir('backup_voc-person'):
basename = osp.splitext(osp.basename(model_fname))[0]
if not basename.startswith(model):
continue
iterations = int(basename.split('_')[1])
max_iterations = max(max_iterations, iterations)
model = '%s_%d' % (model, max_iterations)
print(model)
out_dir = 'out_' + model
if not osp.exists(out_dir):
os.makedirs(out_dir)
for fname in open('/home/wkentaro/data/voc-person/2007_test.txt'):
fname = fname.strip()
print(fname)
cmd = './darknet detect cfg/{0}.cfg backup_voc-person/{1}.weights {2} -i 7'.format(model.split('_')[0], model, fname)
print('+ ' + cmd)
subprocess.call(shlex.split(cmd))
shutil.move('/home/wkentaro/darknet/predictions.png', osp.join(out_dir, osp.basename(fname)))