testing.py 文件源码

def writeResults(DestinationFolder,resultFolder,name_array,classifier,Y_predicted):
    print(len(name_array))
    if not os.path.exists(resultFolder):
        os.mkdir(resultFolder)
    size_m = 0
    i = 0
    j = 0
    lc = 0
    while size_m < len(name_array):
        current = cv2.imread(DestinationFolder+name_array[size_m]+"_final_candidates.bmp")
        #print(DestinationFolder+name_array[size_m]+"_final_candidates.bmp.bmp")        
        #print("current ka size",current.shape)
        x,current_m,z = cv2.split(current)
        #print(count_ones(current_m,255),"now again check",name_array[size_m])
        i = 0       
        while i < current_m.shape[0]:
            j = 0
            while j < current_m.shape[1]:
                if current_m[i,j] == 255:
                    current_m[i,j] = 255*Y_predicted[lc]
                    lc = lc + 1
                j = j + 1
            i = i + 1
        cv2.imwrite(resultFolder+name_array[size_m]+classifier+"_result.bmp",current_m)
        size_m = size_m + 1