wkentaro
2/16/2017 - 7:53 AM

run_tiny.py

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