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