featureExtraction.py 文件源码

def getFeaturesFromFile(fileName, PLOT = False):
    img = cv2.imread(fileName, cv2.CV_LOAD_IMAGE_COLOR)    # read image
    #img2 = resizeFrame(img, 128)# resize

    #img2[:,:,0] = img2[:,:,0] + 3.5 * img2.std() * np.random.random([img2.shape[0], img2.shape[1]])
    #img2[:,:,1] = img2[:,:,1] + 3.5 * img2.std() * np.random.random([img2.shape[0], img2.shape[1]])
    #img2[:,:,2] = img2[:,:,2] + 3.5 * img2.std() * np.random.random([img2.shape[0], img2.shape[1]])    
    #plt.imshow(img2)
    #plt.show()    
    #[F, N] = featureExtraction(img2, PLOT)            # feature extraction
    [F, N] = featureExtraction(img, PLOT)            # feature extraction
    return F, N