int main() {
img0 = cvLoadImage("test.jpg", -1);
data1 = (uchar *) img0->imageData;
height = img0->height;
width = img0->width;
step = (img0->widthStep);
channels = (img0->nChannels);
cvNamedWindow("image0", 1);
for (i = 0; i < img0->height; i++) {
for (j = 0; j < img0->width; j++) {
gray = (int) (0.114 * data1[i * step + j * channels + 0]) +
(0.587 * data1[i * step + j * channels + 1]) +
(0.299 * data1[i * step + j * channels + 2]);
data1[i * step + j * channels + 0] = gray;
data1[i * step + j * channels + 1] = gray;
data1[i * step + j * channels + 2] = gray;
}
}
for (i = 0; i < img0->height; i++) {
for (j = 0; j < img0->width; j++) {
x = (i - (img0->height) / 2.0);
y = (j - (img0->width) / 2.0);
// phi[i][j]=900.0/(900.0+x*x+y*y)-0.5;
phi[i][j] = -sqrt((x - 20) * (x - 20) + (y - 20) * (y - 20)) + 30;
}
}
ChanVeseSegmentation();
for (i = 0; i < img0->height; i++) {
for (j = 0; j < img0->width; j++) {
// if((phi[i][j])>0)
if (phi[i][j] < 0 && phi[i + 1][j] > 0 || phi[i][j] > 0 && phi[i][j - 1] < 0 || phi[i][j] < 0 && phi[i + 1][j - 1] > 0 || phi[i][j] > 0 && phi[i][j + 1] < 0) {
data1[i * step + j * channels + 0] = 0;
data1[i * step + j * channels + 1] = 0;
data1[i * step + j * channels + 2] = 255;
}
}
}
cvShowImage("image0", img0);
cvWaitKey(0);
cvReleaseImage(&img0);
return (0);
} // main end
int main()
{
cout << "Reading in image...\n";
Image<unsigned char>* img = ReadP5image("test.pgm");
Image<unsigned char>* imageOut = new Image<unsigned char>(img->nRows(),
img->nCols());
Image<double>* phi0 = new Image<double>(img->nRows(),
img->nCols());
Image<double>* phi = new Image<double>(img->nRows(),
img->nCols());
Image<unsigned char>* edges = new Image<unsigned char>(img->nRows(),
img->nCols());
// Set up parameters (don't hardcode this!)
struct CVsetup* pCVinputs = new struct CVsetup;
pCVinputs->dt = 0.1;
pCVinputs->h = 1.0;
pCVinputs->lambda1 = 1.0;
pCVinputs->lambda2 = 1.0;
pCVinputs->mu = 0.5;
pCVinputs->nu = 0;
pCVinputs->p = 1;
// Set up initial circular contour for a 256x256 image
double x;
double y;
for (unsigned int i = 0; i < img->nRows(); ++i)
{
for (unsigned int j = 0; j < img->nCols(); ++j)
{
x = double(i) - img->nRows()/2.0;
y = double(j) - img->nCols()/2.0;
phi0->data()[i][j] = 900.0/(900.0 + x*x + y*y) - 0.5;
}
}
cout << "Performing segmentation...\n";
ChanVeseSegmentation(img,phi0, &phi, pCVinputs);
cout << "Getting zero crossings...\n";
ZeroCrossings(phi,&edges,(unsigned char)255,(unsigned char)0);
imageOut->CopyFrom(img);
for (unsigned int i = 0; i < img->nRows(); ++i)
{
for (unsigned int j = 0; j < img->nCols(); ++j)
{
if (edges->data()[i][j] == (unsigned char)255)
imageOut->data()[i][j] = (unsigned char)255;
}
}
cout << "Writing output...\n";
WriteP5image(imageOut,"CVoutEdges.pgm",255);
imageOut->CopyFrom(img);
for (unsigned int i = 0; i < img->nRows(); ++i)
{
for (unsigned int j = 0; j < img->nCols(); ++j)
{
if (phi->data()[i][j] >= 0)
imageOut->data()[i][j] = (unsigned char)0;
else
imageOut->data()[i][j] = (unsigned char)255;
}
}
WriteP5image(imageOut,"CVoutRegions.pgm",255);
cout << "Done.\n";
}
