2012-12-17 15 views
5
#include <cv.h> 
#include <highgui.h> 
#include <iostream> 
#include <cmath> 
#include <cstdlib> 
#include <fstream> 

using namespace std; 

typedef struct histBundle { 
double rCh[256]; 
double gCh[256]; 
double bCh[256]; 
}bundleForHist; 

bundleForHist getHistFromImage (IplImage* img, int numBins) { 
float range[] = { 
    0, 
    numBins 
}; 
float *ranges[] = { 
    range 
}; 

bundleForHist bfh; 

CvHistogram *hist = cvCreateHist (1, &numBins, CV_HIST_ARRAY, ranges, 1); 
cvClearHist (hist); 
IplImage* imgRed = cvCreateImage(cvGetSize(img), 8, 1); 
IplImage* imgGreen = cvCreateImage(cvGetSize(img), 8, 1); 
IplImage* imgBlue = cvCreateImage(cvGetSize(img), 8, 1); 
cvSplit(img, imgBlue, imgGreen, imgRed, NULL); 
cvCalcHist(&imgRed, hist, 0, 0); 
for (int i = 0; i < numBins; ++i) { 
    bfh.rCh[i] = cvQueryHistValue_1D(hist, i); 
} 
cvClearHist(hist); 
cvCalcHist(&imgGreen, hist, 0, 0); 
for (int i = 0; i < numBins; ++i) { 
    bfh.gCh[i] = cvQueryHistValue_1D(hist, i); 
} 
cvClearHist(hist); 
cvCalcHist(&imgBlue, hist, 0, 0); 
for (int i = 0; i < numBins; ++i) { 
    bfh.bCh[i] = cvQueryHistValue_1D(hist, i); 
} 
cvClearHist(hist); 
return bfh; 
} 

int main (int argc, char** argv) { 
int c; 
IplImage* img = NULL; 
int frame_number = 0; 
CvCapture* capture = cvCaptureFromAVI ("Cricketc1.avi"); 
assert(capture); 
int fps = (int)cvGetCaptureProperty(capture, CV_CAP_PROP_FPS); 
cvNamedWindow ("Video", 0); 

while (1) { 
    //IplImage * img = cvLoadImage("C:\\Users\\ANIMES~1\\Desktop\\bw.jpg"); 
    img = cvQueryFrame(capture); 
    frame_number++; 
    if (img) { 
     cvShowImage("Video", img); 
     int numBins = 256; 
     bundleForHist bfh; 
     bfh = getHistFromImage (img, numBins); 
     double totalForR = 0; 
     double totalForG = 0; 
     double totalForB = 0; 
     double probR[256]; 
     double probG[256]; 
     double probB[256]; 
     for (int i = 0; i < numBins-1; ++i) { 
      totalForR += bfh.rCh[i]; 
      totalForG += bfh.gCh[i]; 
      totalForB += bfh.bCh[i]; 
     } 
     double lengthHistogram = totalForR + totalForG + totalForB; 
     for (int i = 0; i < 256; ++i) { 
      probR[i] = bfh.rCh[i]/(double)lengthHistogram; 
      probG[i] = bfh.gCh[i]/(double)lengthHistogram; 
      probB[i] = bfh.bCh[i]/(double)lengthHistogram; 
      //file << bfh.rCh[i] << "\t" << bfh.gCh[i] << "\t" << bfh.bCh[i] << "\t" << probR[i] << "\t" << probG[i] << "\t" << probB[i] << "\n"; 
     } 

     double entropyR = 0.0; 
     double entropyG = 0.0; 
     double entropyB = 0.0; 
     for (int i = 0; i < numBins; ++i) { 
      entropyR += probR[i]*log(probR[i]); 
      entropyG += probG[i]*log(probG[i]); 
      entropyB += probB[i]*log(probB[i]); 
     } 
     cout << frame_number << "\t" << (-1.0)*(entropyR + entropyG + entropyB) << endl; 
    } 
    c = cvWaitKey(1000/fps); 
    if (c == 27) 
     break; 
} 
//cvReleaseImage(&img); 
cvReleaseCapture(&capture); 
cvDestroyWindow("Video"); 
return 0; 
} 

WYJŚCIE:wyjście Weird przy poszukiwaniu entropię klatek wideo w OpenCV

. 
. 
254  -1.#IND 
255  -1.#IND 
256  -1.#IND 
257  -1.#IND 
258  5.5686 
. 
. 

raz pierwszy znalazł entropii obrazu, który wyszedł za prawidłowe. , ale prawie 80% entropii klatek w filmie pochodzi z -1.#IND.

To wideo .... download

Co może być nie tak?

+0

2 sekundy patrząc na kod pokazuje, że robisz "entropia = bla". To nie jest sposób obliczania entropii, musisz zrobić sumę, więc chcesz "entropia + = bla". – mmgp

+0

Oh! Przepraszam ... Zrobiłem to, ale nadal otrzymuję wynik jako '-1. # QNAN' ..... –

+0

Więcej 2 sekund na zaglądanie i widzę, że zwracasz zmienną lokalną na' calcHistProb' i traktując go jako wskaźnik później w twoim programie. Naprawdę wątpię, żebyś skompilował to bez ostrzeżeń i ignorujesz ostrzeżenia. Napraw wszystkie problemy językowe bez ignorowania ostrzeżeń i edytuj swój kod. – mmgp

Odpowiedz

3

Może być tak, że prob[i] = 0 dla niektórych , dlatego obliczasz log(0), który jest niezdefiniowany. Aby ustalić, które po prostu odrzucić takie „prawdopodobieństwo”:

for (int i = 0; i < numBins; ++i) { 
    if (prob[i]) 
     entropy += prob[i]*log(prob[i]); 
} 

Dla innego błędu znalezionego w odniesieniu do wartości w kosza 255 0, to z powodu zakresie określonym przez użytkownika. OpenCV uważa zakres odpowiedniej funkcji za [początek, koniec], więc określenie zakresu [0, 255] będzie ignorować końcową wartość 255. To, czego chcesz, to zachować zarówno 0, jak i 255, dlatego:

float range[] = {0, numBins}; 
+0

Ten problem z obrazem został rozwiązany .... ale mam trudność z wideo ..... –

+0

Zrób to szybciej niż. – mmgp

+0

Rozwiązałem problem z wartością 0 dla obrazu, pytam go, w jaki sposób może wystąpić błąd w przypadku wideo .... ??? –

Powiązane problemy