#!/usr/bin/env python import sys import operator import json import argparse from skimage.util import img_as_float from utils import parse_synonyms, format_classifications classifications = format_classifications(sys.stdin.read()) sorted_list = sorted(classifications.items(), key=operator.itemgetter(1), reverse=True) cl = sorted_list[0][0] pr = classifications[cl] if len(sorted_list) > 1: cl2 = sorted_list[1][0] pr2 = classifications[cl2] if pr2 * 2.0 >= pr: print "Classification not confident." else: print cl + ": " + str(pr) else: print cl + ": " + str(pr)
import os import shutil from skimage import io from skimage.util import img_as_float from utils import format_classifications, parse_synonyms match_dir = "tmp/match" if os.path.isdir(match_dir): shutil.rmtree(match_dir) os.mkdir(match_dir) data = sys.stdin.read() print data classifications = format_classifications(data) sorted_list = sorted(classifications.items(), key=operator.itemgetter(1), reverse=True) cl = sorted_list[0][0] print "searching for " + cl inverted_annotation = json.load(open("tmp/inverted-annotation.json")) annotation = json.load(open("tmp/annotation.json")) log = open("tmp/match/log.txt", "w") class_file = open("tmp/match/class.txt", "w") class_file.write(cl) class_file.close()