Exemplo n.º 1
0
            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
Exemplo n.º 2
0
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');