featuresDOF.py 文件源码

def scriptDemo():
    dirName = "demoData/dofDemo"
    types = ('*.jpg', )
    imageFilesList = []
    for files in types:
        imageFilesList.extend(glob.glob(os.path.join(dirName, files)))

    imageFilesList = sorted(imageFilesList)
    print imageFilesList
    Features = numpy.zeros( (len(imageFilesList), 7) )
    for i, f in enumerate(imageFilesList):
        img = cv2.imread(f, cv2.CV_LOAD_IMAGE_COLOR)    # read image
        [F, names] = getDepthOfFieldFeature2(img)
        Features[i,:] = F
    # Features[i,j] contains the j-th feature of the i-th file
    for i in range(7):
        plt.subplot(7,1,i+1); plt.bar(range(Features[:,i].shape[0]), Features[:,i])
        plt.title(names[i])
    plt.show()
    print Features
#scriptDemo()