facE/start_evaluation.py at master · phmasek/facE · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
from optparse import OptionParser
from apievaluation import apievaluation
import json
import sys
import settings


def main(argv):


    usage = "usage: %prog [options] arg"
    parser = OptionParser(usage)
    parser.add_option("-n", "--noimages", dest="no_images",
                      help="number images to parse", metavar="integer")
    parser.add_option("-m", "--mode", dest="mode",
                      help="Pass what mode to run in", metavar="string")
    parser.add_option("-d", "--directory", dest="directory",
                      help="Pass the directory of the test data", metavar="string")

    (options, args) = parser.parse_args()

    if (options.no_images is None) or (len(options.no_images) == 0):
        parser.error("Missing argument -n or -n is less than 1")
    elif int(options.no_images) > 0 and (options.directory is not None):
        result = apievaluation.initiate_test(int(options.no_images), options.directory, True)
        file = open(settings.ROOT_DIR+'/res/result.json', 'w+')
        file.write(json.dumps(result, sort_keys=True, indent=4, separators=(',', ': ')))
        print 'Wrote result to "%s" and apievaluation database' % settings.ROOT_DIR+'/res/result.json'
    elif int(options.no_images) > 0 and options.mode is not None:
        result = apievaluation.initiate_test(int(options.no_images), options.mode, False)
        file = open(settings.ROOT_DIR+'/res/result.json', 'w+')
        file.write(json.dumps(result, sort_keys=True, indent=4, separators=(',', ': ')))
        print 'Wrote result to "%s"' % settings.ROOT_DIR+'/res/result.json'
    elif int(options.no_images) > 0 and (options.mode is None):
        result = apievaluation.initiate_test(int(options.no_images), None, False)
        file = open(settings.ROOT_DIR+'/res/result.json', 'w+')
        file.write(json.dumps(result, sort_keys=True, indent=4, separators=(',', ': ')))
        print 'Wrote result to "%s"' % (settings.ROOT_DIR+'/res/result.json')
    else:
        parser.error("Argument -n should be more than %d" % int(options.no_images))


if __name__ == "__main__":
    main(sys.argv)