i = randRange[j]; if(not imListAll[i].endswith(".svg")): continue; imList.append(imListAll[i]); for i in range(len(imList)): imList[i] = READ_OUT_WWW + d + "/" + imList[i]; random.shuffle(imList); res[d] = partition(imList,NUM_PARTITIONS_PER_CLASS); return res; def reorder(arr,order): return [arr[i] for i in order]; doneHIT = set(HIT_IO.read_done_HIT(HIT_OUT)); classNames = os.listdir(READ_SOURCE_DIR); classIms = getClassIms(classNames); partNames = getPartNames(classNames); # only get already existing HITS; for no-parts experiment ##for c in classIms: ## partitionTemp = []; ## for partitionImgs in classIms[c]: ## temp = []; ## for img in partitionImgs: ## if img in doneHIT: ## temp.append(img); ## partitionTemp.append(temp); ## classIms[c] = partitionTemp; # only get already existing HITS; for no-parts experiment
import time; import HIT_IO; import HIT_parser; import Stroke_parser; RES_DIR = 'results/noParts2/'; RES_ANNOT_DIR = RES_DIR+ 'annotations/'; OUT_FILE = RES_DIR + 'allResults.csv'; HIT_OUT = RES_DIR + "HIT_OUT"; HIT_parser.compile_HIT_results(RES_DIR, OUT_FILE); res = HIT_parser.parse_HIT_lines(OUT_FILE); mapping = res['mapping']; parsed = res['parsed']; HIT_IO.output_done_HIT(mapping,parsed,HIT_OUT); annotOnlyPref = 'annotOnly-'; # json strokes totalPref = ''; # svg overlay svgAnnotePref = 'annotOnly-'; # svg annotation only classMatch = '/(.+?)/(\d+)\.svg'; imNumCounts = {}; for imgs in parsed: strokes = Stroke_parser.parse_strokes(imgs[mapping['Answer.strokes']]); svgMatch = '(svgSubset/.+)'; svgName = re.search(svgMatch,imgs[mapping['Input.image_url']]).group(0); print svgName svgFile = open(svgName,'r');