Ejemplo n.º 1
0
    def test_copyToNpArray(self):
        floatVec = hogc.makeTestFloatVec()
        npFloatVec = np.zeros((len(floatVec)+2), dtype=np.float32)
        hogc.copyToNpArray(floatVec, 2, npFloatVec)
        for i in range(0, len(floatVec)):
            self.assertTrue(almostEqual(npFloatVec[i+2], floatVec[i]))
            pass


        pass
Ejemplo n.º 2
0
def hog(img, hogScheme, aHistTemplate, smAngle, smMag, descriptor):
    #print img.width, img.height
    imgCopy = cv2.resize(img, (128, 64))
    numpyImg = np.asarray(imgCopy)
    hogc.gammaCorrect(numpyImg, 2.2, True)
    #normalizedImg = normalizeImage(numpyImg)
    convertedImg = cv2.cvtColor( numpyImg, cv2.COLOR_BGR2GRAY)
    h = convertedImg.shape[0]
    w = convertedImg.shape[1]
    assert(descriptor.shape[0] == hogScheme.numBlocksVertical *  hogScheme.numBlocksHorizontal * hogScheme.numCellsInUnitPerSide * hogScheme.numCellsInUnitPerSide * 9)
    assert(descriptor.dtype == np.float32)
    hog = hogc.Hog(hogScheme, hogc.Hog.BlockGaussianWeightingPolicy.YesGaussian, hogc.Hog.BlockNormalizationPolicy.L2Normalization)
    for r in range(0, hogScheme.numBlocksVertical):
        for c in range(0, hogScheme.numBlocksHorizontal):
            histVec = hogc.FloatVec()
            hog.computeBlock( convertedImg,aHistTemplate,  r, c, smAngle, smMag, False, histVec)
            hogc.copyToNpArray(histVec, (r*15 +c)*36, descriptor)
    return