void annTrain(Mat TrainData, Mat classes, int nNeruns)
{
ann.clear();
Mat layers(1, 3, CV_32SC1);
layers.at<int>(0) = TrainData.cols;
layers.at<int>(1) = nNeruns;
layers.at<int>(2) = numAll;
ann.create(layers, CvANN_MLP::SIGMOID_SYM, 1, 1);
//Prepare trainClases
//Create a mat with n trained data by m classes
Mat trainClasses;
trainClasses.create( TrainData.rows, numAll, CV_32FC1 );
for( int i = 0; i < trainClasses.rows; i++ )
{
for( int k = 0; k < trainClasses.cols; k++ )
{
//If class of data i is same than a k class
if( k == classes.at<int>(i) )
trainClasses.at<float>(i,k) = 1;
else
trainClasses.at<float>(i,k) = 0;
}
}
Mat weights( 1, TrainData.rows, CV_32FC1, Scalar::all(1) );
//Learn classifier
ann.train( TrainData, trainClasses, weights );
}
void annTrain(Mat TrainData, Mat classes, int nNeruns) {
ann.clear();
Mat layers(1, 3, CV_32SC1);
layers.at<int>(0) = TrainData.cols;
layers.at<int>(1) = nNeruns;
layers.at<int>(2) = numAll;
ann.create(layers, CvANN_MLP::SIGMOID_SYM, 1, 1);
// Prepare trainClases
// Create a mat with n trained data by m classes
Mat trainClasses;
trainClasses.create(TrainData.rows, numAll, CV_32FC1);
for (int i = 0; i < trainClasses.rows; i++) {
for (int k = 0; k < trainClasses.cols; k++) {
// If class of data i is same than a k class
if (k == classes.at<int>(i))
trainClasses.at<float>(i, k) = 1;
else
trainClasses.at<float>(i, k) = 0;
}
}
Mat weights(1, TrainData.rows, CV_32FC1, Scalar::all(1));
// Learn classifier
// ann.train( TrainData, trainClasses, weights );
// Setup the BPNetwork
// Set up BPNetwork's parameters
CvANN_MLP_TrainParams params;
params.train_method = CvANN_MLP_TrainParams::BACKPROP;
params.bp_dw_scale = 0.1;
params.bp_moment_scale = 0.1;
// params.train_method=CvANN_MLP_TrainParams::RPROP;
// params.rp_dw0 = 0.1;
// params.rp_dw_plus = 1.2;
// params.rp_dw_minus = 0.5;
// params.rp_dw_min = FLT_EPSILON;
// params.rp_dw_max = 50.;
ann.train(TrainData, trainClasses, Mat(), Mat(), params);
}
int main (int argc, char** argv){
CvANN_MLP* neuron = NULL ;
IplImage *img = cvLoadImage(argv[1], CV_LOAD_IMAGE_COLOR);
CvMat *input3Ch = cvCreateMat(img->height, img->width, CV_32FC3);
CvMat *output1Ch = cvCreateMat(img->height, img->width, CV_32FC1);
CvMat *input_morph = cvCreateMat(img->height, img->width, CV_32FC1);
//cvConvertScale(img, output1Ch);
cvConvertScale(img, input3Ch);
if (neuron == NULL )
neuron = new CvANN_MLP();
else
neuron->clear();
ByteArrayToANN("cia.tmp", neuron);
CvMat input_nn = cvMat(input3Ch->height*input3Ch->width, 3, CV_32FC1, input3Ch->data.fl);
CvMat output_nn = cvMat(output1Ch->height*output1Ch->width, 1, CV_32FC1, output1Ch->data.fl);
neuron->predict(&input_nn, &output_nn);
//do threshold
cvThreshold(output1Ch, input_morph, -0.5, 255.0, CV_THRESH_BINARY);
// morph open
IplConvKernel *se1 = cvCreateStructuringElementEx(3, 3, 1, 1, CV_SHAPE_ELLIPSE);
cvMorphologyEx(input_morph, input_morph, NULL, se1, CV_MOP_OPEN); // remove noise
CvMat *out_single = cvCreateMat(input_morph->height, input_morph->width, CV_32FC1);
cvSetZero(out_single);
IplImage *tmp8UC1 = cvCreateImage(cvGetSize(input_morph), IPL_DEPTH_8U, 1);
// remove small cells and fill holes.
CvMemStorage *storage = cvCreateMemStorage();
CvSeq *first_con = NULL;
CvSeq *cur = NULL;
cvConvert(input_morph, tmp8UC1);
cvFindContours(tmp8UC1, storage, &first_con, sizeof(CvContour), CV_RETR_EXTERNAL);
cur = first_con;
while (cur != NULL) {
double area = fabs(cvContourArea(cur));
int npts = cur->total;
CvPoint *p = new CvPoint[npts];
cvCvtSeqToArray(cur, p);
//~ cout<<area<<" ";
if (area < 1500.0) // remove small area
cvFillPoly(input_morph, &p, &npts, 1, cvScalar(0.0)); // remove from input
else if (area < 7500.0) {
cvFillPoly(out_single, &p, &npts, 1, cvScalar(255.0)); // move to single
cvFillPoly(input_morph, &p, &npts, 1, cvScalar(0.0)); // remove from input
}else
cvFillPoly(input_morph, &p, &npts, 1, cvScalar(255.0)); // fill hole
delete[] p;
cur = cur->h_next;
}
//~ cout<<endl;
//~ Mat tmpmat = cvarrToMat(out_single, true);
//~ tmpmat.convertTo(tmpmat, CV_8UC1);
//~ if(!imwrite("c_out_single.jpg", tmpmat))
//~ {
//~ cout<<"error writing image"<<endl;
//~ }
//~
//~ tmpmat = cvarrToMat(input_morph, true);
//~ tmpmat.convertTo(tmpmat, CV_8UC1);
//~ if(!imwrite("c_input_morph.jpg", tmpmat))
//~ {
//~ cout<<"error writing image"<<endl;
//~ }
//~
//~ cvSaveImage("c_tmp8UC1.jpg", tmp8UC1);
CvMat *output_morph = cvCreateMat(input_morph->height, input_morph->width, CV_32FC1);
cvOr(input_morph, out_single, output_morph);
cvReleaseStructuringElement(&se1);
//~ tmpmat = cvarrToMat(output_morph, true);
//~ tmpmat.convertTo(tmpmat, CV_8UC1);
//~ if(!imwrite("c_output_morph_or.jpg", tmpmat))
//~ {
//~ cout<<"error writing image"<<endl;
//~ }
//## Scanning Cells ##//
int ncell = 0, prev_ncontour = 0, same_count = 0, ncontour = 1;
while(ncontour != 0){
cvConvert(input_morph, tmp8UC1);
cvClearMemStorage(storage);
int ncontour = cvFindContours(tmp8UC1, storage, &first_con, sizeof(CvContour), CV_RETR_EXTERNAL);
if (ncontour == 0)
break; // finish extract cell
if (ncontour == prev_ncontour) {
cvErode(input_morph, input_morph);
same_count++;
} else
same_count = 0;
prev_ncontour = ncontour;
cur = first_con;
while (cur != NULL) {
double area = fabs(cvContourArea(cur));
if ((area < 3000.0) || (same_count > 10)) {
int npts = cur->total;
CvPoint *p = new CvPoint[npts];
cvCvtSeqToArray(cur, p);
cvFillPoly(out_single, &p, &npts, 1, cvScalar(255.0)); // move to single
cvFillPoly(input_morph, &p, &npts, 1, cvScalar(0.0)); // remove from input
delete[] p;
ncell++;
}
cur = cur->h_next;
}
}
Mat tmpmat = cvarrToMat(out_single, true);
//tmpmat.convertTo(tmpmat, CV_8UC1);
if(!saveMat("outSingle", tmpmat))
{
cout << "outSingle: cant save mat to binary file" << endl;
}
tmpmat = cvarrToMat(output_morph, true);
//tmpmat.convertTo(tmpmat, CV_8UC1);
if(!saveMat("outputMorph", tmpmat))
{
cout << "outMorph : save mat to binary file" << endl;
}
//~ if(!imwrite("c_out_single_scanningCell.jpg", tmpmat))
//~ {
//~ cout<<"error writing image"<<endl;
//~ }
//~
//~
//~ tmpmat = cvarrToMat(input_morph, true);
//~ tmpmat.convertTo(tmpmat, CV_8UC1);
//~ if(!imwrite("c_input_morph_scanningCell.jpg", tmpmat))
//~ {
//~ cout<<"error writing c_out_single_scanningCell.jpg"<<endl;
//~ }
//~
//~ tmpmat = cvarrToMat(output_morph, true);
//~ tmpmat.convertTo(tmpmat, CV_8UC1);
//~ if(!imwrite("c_output_morph_scanningCell.jpg", tmpmat))
//~ {
//~ cout<<"error writing c_output_morph_scanningCell.jpg"<<endl;
//~ }
//~ cvSaveImage("c_tmp8UC1_scanningCell.jpg", tmp8UC1);
/* ### separate cells ### */
cvConvert(out_single, tmp8UC1);
cvClearMemStorage(storage);
cvFindContours(tmp8UC1, storage, &first_con, sizeof(CvContour), CV_RETR_EXTERNAL);
int count = 1;
cur = first_con;
while (cur != NULL) {
int npts = cur->total;
CvPoint *p = new CvPoint[npts];
cvCvtSeqToArray(cur, p);
cvFillPoly(out_single, &p, &npts, 1, cvScalar((count++%254)+1)); // fill label, must be 1-255
delete[] p;
cur = cur->h_next;
}
cvConvertScale(output_morph, tmp8UC1);
CvMat *inwater = cvCreateMat(out_single->height, out_single->width, CV_8UC3);
CvMat outwater = cvMat(out_single->height, out_single->width, CV_32SC1, out_single->data.fl);
cvMerge(tmp8UC1, tmp8UC1, tmp8UC1, NULL, inwater);
cvWatershed(inwater, &outwater);
cvErode(out_single, out_single, NULL, 2);
cvConvertScale(output_morph, tmp8UC1);
cvSub(output_morph, out_single, output_morph, tmp8UC1);
cvReleaseMat(&inwater);
//~
//~ tmpmat = cvarrToMat(out_single, true);
//~ tmpmat.convertTo(tmpmat, CV_8UC1);
//~ if(!imwrite("c_out_single_sep.jpg", tmpmat))
//~ {
//~ cout<<"error writing c_out_single_sep.jpg"<<endl;
//~ }
//~
//~
//~ tmpmat = cvarrToMat(input_morph, true);
//~ tmpmat.convertTo(tmpmat, CV_8UC1);
//~ if(!imwrite("c_input_morph_sep.jpg", tmpmat))
//~ {
//~ cout<<"error writing c_input_morph_sep.jpg"<<endl;
//~ }
//~
//~ cvSaveImage("c_tmp8UC1_sep.jpg", tmp8UC1);
//~
//~ tmpmat = cvarrToMat(output_morph, true);
//~ tmpmat.convertTo(tmpmat, CV_8UC1);
//~ if(!imwrite("c_output_morph_sep.jpg", tmpmat))
//~ {
//~ cout<<"error writing c_output_morph_sep.jpg"<<endl;
//~ }
/* ### prepare result ### */
cvConvertScale(output_morph, tmp8UC1);
cvClearMemStorage(storage);
cvFindContours(tmp8UC1, storage, &first_con, sizeof(CvContour), CV_RETR_EXTERNAL);
IplImage *tmpImage = cvCreateImage(cvSize(tmp8UC1->width, tmp8UC1->height), IPL_DEPTH_8U, 3);
cvSet(tmpImage, CV_RGB(0,0,255)); // Background, blue
cvSetZero(tmp8UC1);
CvScalar pixel;
cur = first_con;
ncell = 0; // total cells
while (cur != NULL) {
if ((cur->total > 2) && (fabs(cvContourArea(cur)) > 1500.0)) { // remove small area
int npts = cur->total;
CvPoint *p = new CvPoint[npts];
cvCvtSeqToArray(cur, p);
cvFillPoly(tmp8UC1, &p, &npts, 1, cvScalar(255)); // set mask
pixel = cvAvg(output1Ch, tmp8UC1);
cvFillPoly(tmp8UC1, &p, &npts, 1, cvScalar(0)); // clear mask
if (pixel.val[0] > 0.5) { // Negative, green
if (tmpImage != NULL)
cvFillPoly(tmpImage, &p, &npts, 1, CV_RGB(0,255,0));
} else if (pixel.val[0] > -0.5) { // Positive, red
if (tmpImage != NULL)
cvFillPoly(tmpImage, &p, &npts, 1, CV_RGB(255,0,0));
}
delete[] p;
}
cur = cur->h_next;
}
cvSaveImage("result_fullCIA.jpg", tmpImage);
cvReleaseMat(&inwater);
//cvReleaseMat(outwater);
if (tmp8UC1 != NULL) cvReleaseImage(&tmp8UC1);
if (input_morph != NULL) cvReleaseMat(&input_morph);
if (out_single != NULL) cvReleaseMat(&out_single);
if (output_morph != NULL) cvReleaseMat(&output_morph);
//if (input3Ch != NULL) cvReleaseMat(&input3Ch);
if (output1Ch != NULL) cvReleaseMat(&output1Ch);
if (tmpImage != NULL) cvReleaseImageHeader(&tmpImage);
if (storage != NULL) cvReleaseMemStorage(&storage);
return 0;
}
int ns__trainANN(struct soap *soap,
char *inputMatFilename,
char *neuralFile,
char **OutputMatFilename)
{
double start, end;
start = omp_get_wtime();
Mat src; //must be 3ch image
if(!readMat(inputMatFilename, src))
{
cerr << "trainANN :: can not read bin file" << endl;
soap->fault->faultstring = "trainANN :: can not read bin file";
return SOAP_FAULT;
}
// convert src to CvMat to use an old-school function
CvMat tmp = src;
CV_Assert(tmp.cols == src.cols && tmp.rows == src.rows &&
tmp.data.ptr == (uchar*)src.data && tmp.step == src.step);
CvMat *input3Ch = cvCreateMat(src.rows, src.cols, CV_32FC3);
cvConvert(&tmp, input3Ch);
CvMat *output1Ch = cvCreateMat(src.rows, src.cols, CV_32FC1);
CvANN_MLP* neuron = NULL ;
if (neuron == NULL )
neuron = new CvANN_MLP();
else
neuron->clear();
if(!ByteArrayToANN(neuralFile, neuron)){
cerr << "trainANN :: can not load byte array to neural" << endl;
return soap_receiver_fault(soap, "trainANN :: can not load byte array to neural", NULL);
}
CvMat input_nn = cvMat(input3Ch->height*input3Ch->width, 3, CV_32FC1, input3Ch->data.fl);
CvMat output_nn = cvMat(output1Ch->height*output1Ch->width, 1, CV_32FC1, output1Ch->data.fl);
neuron->predict(&input_nn, &output_nn);
Mat resultNN = cvarrToMat(output1Ch, false);
/* generate output file name */
*OutputMatFilename = (char*)soap_malloc(soap, FILENAME_SIZE);
time_t now = time(0);
strftime(*OutputMatFilename, sizeof(OutputMatFilename)*FILENAME_SIZE, "/home/lluu/dir/%Y%m%d_%H%M%S_trainANN", localtime(&now));
/* save to bin */
if(!saveMat(*OutputMatFilename, resultNN))
{
cerr << "trainANN :: save mat to binary file" << endl;
return soap_receiver_fault(soap, "trainANN :: save mat to binary file", NULL);
}
src.release();
resultNN.release();
cvReleaseMat(&output1Ch);
end = omp_get_wtime();
cerr<<"ns__trainANN "<<"time elapsed "<<end-start<<endl;
return SOAP_OK;
}
