void CmSaliencyGC::GetGU(vecD& gc, vecD &d, double sigmaDist, double dominate) { if (d.size() == 0){ d.resize(_NUM); #pragma omp parallel for for (int i = 0; i < _NUM; i++) SpatialVar(_PixelSalCi1f[i], d[i]); } Mat_<double> clrDist; clrDist = Mat_<double>::zeros(_NUM, _NUM); vector<Vec3f> gmmClrs(_NUM); cvtColor(_gmmClrs, gmmClrs, CV_BGR2Lab); vecD maxDist(_NUM); for (int i = 0; i < _NUM; i++) for (int j = 0; j < i; j++){ double dCrnt = vecDist(gmmClrs[i], gmmClrs[j]); clrDist(i, j) = clrDist(j, i) = dCrnt; maxDist[i] = max(maxDist[i], dCrnt); maxDist[j] = max(maxDist[j], dCrnt); } for (int i = 0; i < _NUM; i++) { gc[i] = 0; for (int j = 0; j < _NUM; j++) gc[i] += _gmmW[j] * clrDist(i, j) / maxDist[i]; //min(clrDist(i, j), dominate); } for (int i = 0; i < _NUM; i++) _gu[i] = _gu[i] * exp(-9.0 * sqr(d[i])); // normalize(_gu, _gu, 0, 1, CV_MINMAX); }
void CmEvaluation::PrintVector(FILE *f, const vecD &v, CStr &name) { fprintf(f, "%s = [", name.c_str()); for (size_t i = 0; i < v.size(); i++) fprintf(f, "%g ", v[i]); fprintf(f, "];\n"); }
void Illustrate::__PrintVector(const vecD &v, CStr &name) { FILE* f = fopen(_S(name), "w"); CV_Assert(f != NULL); for (size_t i = 0; i < v.size(); i++) { fprintf(f, "%g\n", v[i]); } fclose(f); }
int Saliency::PrintVector(FILE *f, const vecD &v, const string &name, int maxNum) { fprintf(f, "%s = [", name.c_str()); maxNum = min(maxNum, (int)v.size()); int i; for (i = 0; i < maxNum; i++) if (v[i] > 0.001) // Very small recall is too much noisy (may due to compression) fprintf(f, "%g ", v[i]); else break; fprintf(f, "];\n"); return i; }
// load format(imgW, i) and add information to the back of imgs and lens void CmIllustr::LoadImgs(CStr &imgW, vecM &imgs, vecD &lens, int W, int H) { bool toRow = W > H; double crnt = -space; if (imgs.size()){ // There exist a predefined image for sketch lens.push_back(toRow ? H*imgs[0].cols*1./imgs[0].rows : W*imgs[0].rows*1./imgs[0].cols); crnt += lens[0] + space; } for (int i = 0; i < 500; i++){ string imgN = format(_S(imgW), i), inDir, maskN; vecS names; int subN = CmFile::GetNames(imgN, names, inDir); if (subN == 0) continue; Mat img = imread(inDir + names[0]); if (img.data == NULL){ printf("Can't load image file %-70s\n", _S(names[0])); continue; } if (subN > 1){ Mat mask1u = imread(inDir + names[1], CV_LOAD_IMAGE_GRAYSCALE), big1u; dilate(mask1u, big1u, Mat(), Point(-1, -1), 5); bitwise_xor(mask1u, big1u, mask1u); img.setTo(Scalar(0, 0, 255), mask1u); } lens.push_back(toRow ? H*img.cols*1./img.rows : W*img.rows*1./img.cols); imgs.push_back(img); crnt += lens[lens.size() - 1] + space; if (crnt >= max(H, W)) break; } int num = imgs.size(); if (num && abs(crnt - max(H,W)) > abs(crnt - lens[num - 1] - space - max(H,W))) imgs.resize(num - 1), lens.resize(num - 1); printf("%s: %d\n", _S(imgW), num); if (crnt < max(H, W)) { printf(_S(imgW + ": not enough images\n")); exit(0); } }
Mat CmIllustr::ArrangeImgs(vecM &imgs, vecD &len, int W, int H, bool toRow) { int imgN = (int)(imgs.size()), s = 0; CV_Assert(len.size() == imgN); double ratio, sumL = 0, err = 0; for (int i = 0; i < imgN; i++) sumL += len[i]; ratio = ((toRow ? W : H) - (imgN - 1) * space) / sumL; Mat dstImg(H, W, CV_8UC3); dstImg = Scalar(255, 255, 255); for (int i = 0; i < imgN; i++) { len[i] *= ratio; int l = cvRound(len[i] + err); Rect reg = toRow ? Rect(s, 0, l, H) : Rect(0, s, W, l); resize(imgs[i], dstImg(reg), reg.size()); err = len[i] + err - l; s += l + space; } CV_Assert(s - space == (toRow ? dstImg.cols : dstImg.rows)); return dstImg; }