#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;
}
OUTPUT:奇怪的輸出,同時在OpenCV中找到一個視頻幀的熵
.
.
254 -1.#IND
255 -1.#IND
256 -1.#IND
257 -1.#IND
258 5.5686
.
.
我第一次發現圖像熵,出來是正確的。 但近80%的視頻幀entropies出來是-1.#IND
。
這是視頻.... download
可能是什麼回事?
查看代碼2秒顯示您正在執行'entropy = blah'。這不是如何計算熵,你需要做一個總和,所以你想要'熵+ =等'。 – mmgp
哦!我很抱歉...我這樣做,但仍然得到輸出作爲'-1。#QNAN' ..... –
更多的2秒看着它,我看到你正在返回一個局部變量'calcHistProb'和在程序中稍後將它作爲指針處理。我真的懷疑你是在沒有得到警告的情況下編譯它的,而且你忽略了警告。修正所有這些語言問題而不忽略警告,並編輯您的代碼。 – mmgp