You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
390 lines
11 KiB
390 lines
11 KiB
2 months ago
|
#ifdef CONFIG_UCW_CLEAN_ABI
|
||
|
|
#define image_signatures_dist ucw_image_signatures_dist
|
||
|
|
#define image_signatures_dist_explain ucw_image_signatures_dist_explain
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#ifdef EXPLAIN
|
||
|
|
# define MSG(x...) do{ line += sprintf(line, x); }while(0)
|
||
|
|
# define LINE do{ line = buf; msg(line, param); }while(0)
|
||
|
|
|
||
|
|
static void explain_signature(struct image_signature *sig, void (*msg)(byte *text, void *param), void *param)
|
||
|
|
{
|
||
|
|
byte buf[1024], *line = buf;
|
||
|
|
MSG("signature: flags=0x%x df=%u dh=%u f=(%u", sig->flags, sig->df, sig->dh, sig->vec.f[0]);
|
||
|
|
for (uint i = 1; i < IMAGE_VEC_F; i++)
|
||
|
|
MSG(" %u", sig->vec.f[i]);
|
||
|
|
MSG(")");
|
||
|
|
LINE;
|
||
|
|
for (uint j = 0; j < sig->len; j++)
|
||
|
|
{
|
||
|
|
struct image_region *reg = sig->reg + j;
|
||
|
|
MSG("region %u: wa=%u wb=%u f=(%u", j, reg->wa, reg->wb, reg->f[0]);
|
||
|
|
for (uint i = 1; i < IMAGE_VEC_F; i++)
|
||
|
|
MSG(" %u", reg->f[i]);
|
||
|
|
MSG(") h=(%u", reg->h[0]);
|
||
|
|
for (uint i = 1; i < IMAGE_REG_H; i++)
|
||
|
|
MSG(" %u", reg->h[i]);
|
||
|
|
MSG(")");
|
||
|
|
LINE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#else
|
||
|
|
# define MSG(x...) do{}while(0)
|
||
|
|
# define LINE do{}while(0)
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#define MSGL(x...) do{ MSG(x); LINE; }while(0)
|
||
|
|
|
||
|
|
#ifndef EXPLAIN
|
||
|
|
static uint image_signatures_dist_integrated(struct image_signature *sig1, struct image_signature *sig2)
|
||
|
|
#else
|
||
|
|
static uint image_signatures_dist_integrated_explain(struct image_signature *sig1, struct image_signature *sig2, void (*msg)(byte *text, void *param), void *param)
|
||
|
|
#endif
|
||
|
|
{
|
||
|
|
uint dist[IMAGE_REG_MAX * IMAGE_REG_MAX], p[IMAGE_REG_MAX], q[IMAGE_REG_MAX];
|
||
|
|
uint n, i, j, k, l, s, d;
|
||
|
|
struct image_region *reg1, *reg2;
|
||
|
|
#ifdef EXPLAIN
|
||
|
|
byte buf[1024], *line = buf;
|
||
|
|
MSGL("Integrated matching");
|
||
|
|
explain_signature(sig1, msg, param);
|
||
|
|
explain_signature(sig2, msg, param);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
/* FIXME: do not mux textured and non-textured images (should be split in clusters tree) */
|
||
|
|
if ((sig1->flags ^ sig2->flags) & IMAGE_SIG_TEXTURED)
|
||
|
|
{
|
||
|
|
MSGL("Textured vs non-textured");
|
||
|
|
return ~0U;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Compute distance matrix */
|
||
|
|
n = 0;
|
||
|
|
MSGL("Distance matrix:");
|
||
|
|
/* ... for non-textured images */
|
||
|
|
if (!((sig1->flags | sig2->flags) & IMAGE_SIG_TEXTURED))
|
||
|
|
for (j = 0, reg2 = sig2->reg; j < sig2->len; j++, reg2++)
|
||
|
|
for (i = 0, reg1 = sig1->reg; i < sig1->len; i++, reg1++)
|
||
|
|
{
|
||
|
|
uint dt = 0, ds = 0, dp = 0, d;
|
||
|
|
for (uint i = 0; i < IMAGE_VEC_F; i++)
|
||
|
|
dt += image_sig_cmp_features_weights[i] * isqr((int)reg1->f[i] - (int)reg2->f[i]);
|
||
|
|
for (uint i = 0; i < 3; i++)
|
||
|
|
ds += image_sig_cmp_features_weights[IMAGE_VEC_F + i] * isqr((int)reg1->h[i] - (int)reg2->h[i]);
|
||
|
|
for (uint i = 3; i < 5; i++)
|
||
|
|
dp += image_sig_cmp_features_weights[IMAGE_VEC_F + i] * isqr((int)reg1->h[i] - (int)reg2->h[i]);
|
||
|
|
#if 0
|
||
|
|
int x1, y1, x2, y2;
|
||
|
|
if (sig1->cols > sig1->rows)
|
||
|
|
{
|
||
|
|
x1 = reg1->h[3];
|
||
|
|
y1 = ((int)reg1->h[4] - 64) * (int)sig1->rows / (int)sig1->cols + 64;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
y1 = reg1->h[4];
|
||
|
|
x1 = ((int)reg1->h[3] - 64) * (int)sig1->cols / (int)sig1->rows + 64;
|
||
|
|
}
|
||
|
|
if (sig2->cols > sig2->rows)
|
||
|
|
{
|
||
|
|
x2 = reg2->h[3];
|
||
|
|
y2 = ((int)reg2->h[4] - 64) * (int)sig2->rows / (int)sig2->cols + 64;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
y2 = reg2->h[4];
|
||
|
|
x2 = ((int)reg2->h[3] - 64) * (int)sig2->cols / (int)sig2->rows + 64;
|
||
|
|
}
|
||
|
|
MSGL("%d %d %d %d", x1, y1, x2, y2);
|
||
|
|
dp = image_sig_cmp_features_weights[IMAGE_VEC_F + 3] * isqr(x1 - x2) +
|
||
|
|
image_sig_cmp_features_weights[IMAGE_VEC_F + 4] * isqr(y1 - y2);
|
||
|
|
#endif
|
||
|
|
#if 0
|
||
|
|
d = dt * (4 + MIN(8, (ds >> 12))) * (4 + MIN(8, (dp >> 10))) + (ds >> 11) + (dp >> 10);
|
||
|
|
MSG("[%u, %u] d=%u=(%u * %u * %u + %u + %u) dt=%u ds=%u dp=%u df=(%d", i, j, d,
|
||
|
|
dt, 4 + MIN(8, (ds >> 12)), 4 + MIN(8, dp >> 10), ds >> 11, dp >> 10, dt, ds, dp, (int)reg1->f[0] - (int)reg2->f[0]);
|
||
|
|
#endif
|
||
|
|
#if 1
|
||
|
|
d = dt;
|
||
|
|
if (ds < 1000)
|
||
|
|
d = d * 4;
|
||
|
|
else if (ds < 4000)
|
||
|
|
d = d * 6 + 8;
|
||
|
|
else if (ds < 10000)
|
||
|
|
d = d * 8 + 20;
|
||
|
|
else if (ds < 50000)
|
||
|
|
d = d * 10 + 50;
|
||
|
|
else
|
||
|
|
d = d * 12 + 100;
|
||
|
|
if (dp < 1000)
|
||
|
|
d = d * 2;
|
||
|
|
else if (dp < 4000)
|
||
|
|
d = d * 3 + 100;
|
||
|
|
else if (dp < 10000)
|
||
|
|
d = d * 4 + 800;
|
||
|
|
else
|
||
|
|
d = d * 5 + 3000;
|
||
|
|
#endif
|
||
|
|
dist[n++] = (d << 8) + i + (j << 4);
|
||
|
|
MSG("[%u, %u] d=%u dt=%u ds=%u dp=%u df=(%d", i, j, d, dt, ds, dp, (int)reg1->f[0] - (int)reg2->f[0]);
|
||
|
|
#ifdef EXPLAIN
|
||
|
|
for (uint i = 1; i < IMAGE_VEC_F; i++)
|
||
|
|
MSG(" %d", (int)reg1->f[i] - (int)reg2->f[i]);
|
||
|
|
MSG(") dh=(%d", (int)reg1->h[0] - (int)reg2->h[0]);
|
||
|
|
for (uint i = 1; i < IMAGE_REG_H; i++)
|
||
|
|
MSG(" %d", (int)reg1->h[i] - (int)reg2->h[i]);
|
||
|
|
MSGL(")");
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
/* ... for textured images (ignore shape properties) */
|
||
|
|
else
|
||
|
|
for (j = 0, reg2 = sig2->reg; j < sig2->len; j++, reg2++)
|
||
|
|
for (i = 0, reg1 = sig1->reg; i < sig1->len; i++, reg1++)
|
||
|
|
{
|
||
|
|
uint dt = 0;
|
||
|
|
for (uint i = 0; i < IMAGE_VEC_F; i++)
|
||
|
|
dt += image_sig_cmp_features_weights[i] * isqr((int)reg1->f[i] - (int)reg2->f[i]);
|
||
|
|
dist[n++] = (dt << 12) + i + (j << 4);
|
||
|
|
#ifdef EXPLAIN
|
||
|
|
MSG("[%u, %u] dt=%u df=(%d", i, j, dt, (int)reg1->f[0] - (int)reg2->f[0]);
|
||
|
|
for (uint i = 1; i < IMAGE_VEC_F; i++)
|
||
|
|
MSG(" %d", (int)reg1->f[i] - (int)reg2->f[i]);
|
||
|
|
MSGL(")");
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
/* One or both signatures have no regions */
|
||
|
|
if (!n)
|
||
|
|
return ~0U;
|
||
|
|
|
||
|
|
/* Get percentages */
|
||
|
|
for (i = 0, reg1 = sig1->reg; i < sig1->len; i++, reg1++)
|
||
|
|
p[i] = reg1->wb;
|
||
|
|
for (i = 0, reg2 = sig2->reg; i < sig2->len; i++, reg2++)
|
||
|
|
q[i] = reg2->wb;
|
||
|
|
|
||
|
|
/* Sort entries in distance matrix */
|
||
|
|
image_signatures_dist_integrated_sort(dist, n);
|
||
|
|
|
||
|
|
/* Compute significance matrix and resulting distance */
|
||
|
|
uint sum = 0;
|
||
|
|
MSGL("Significance matrix:");
|
||
|
|
for (k = 0, l = 128; l; k++)
|
||
|
|
{
|
||
|
|
i = dist[k] & 15;
|
||
|
|
j = (dist[k] >> 4) & 15;
|
||
|
|
d = dist[k] >> 8;
|
||
|
|
if (p[i] <= q[j])
|
||
|
|
{
|
||
|
|
s = p[i];
|
||
|
|
q[j] -= p[i];
|
||
|
|
p[i] = 0;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
s = q[j];
|
||
|
|
p[i] -= q[j];
|
||
|
|
q[j] = 0;
|
||
|
|
}
|
||
|
|
l -= s;
|
||
|
|
sum += s * d;
|
||
|
|
#ifdef EXPLAIN
|
||
|
|
reg1 = sig1->reg + i;
|
||
|
|
reg2 = sig2->reg + j;
|
||
|
|
MSG("[%u, %u] s=%u d=%u df=(%d", i, j, s, d, (int)reg1->f[0] - (int)reg2->f[0]);
|
||
|
|
for (uint i = 1; i < IMAGE_VEC_F; i++)
|
||
|
|
MSG(" %d", (int)reg1->f[i] - (int)reg2->f[i]);
|
||
|
|
if (!((sig1->flags | sig2->flags) & IMAGE_SIG_TEXTURED))
|
||
|
|
{
|
||
|
|
MSG(") dh=(%d", (int)reg1->h[0] - (int)reg2->h[0]);
|
||
|
|
for (uint i = 1; i < IMAGE_REG_H; i++)
|
||
|
|
MSG(" %d", (int)reg1->h[i] - (int)reg2->h[i]);
|
||
|
|
}
|
||
|
|
MSGL(")");
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
d = sum / 32;
|
||
|
|
|
||
|
|
uint a = sig1->cols * sig2->rows;
|
||
|
|
uint b = sig1->rows * sig2->cols;
|
||
|
|
if (a < 2 * b && b < 2 * a)
|
||
|
|
d = d * 2;
|
||
|
|
else if (a < 4 * b && b < 4 * a)
|
||
|
|
d = d * 3;
|
||
|
|
else
|
||
|
|
d = d * 5;
|
||
|
|
|
||
|
|
a = sig1->cols * sig1->rows;
|
||
|
|
b = sig2->cols * sig2->rows;
|
||
|
|
|
||
|
|
if ((a < 1000 && b > 5000) || (b < 1000 && a > 5000))
|
||
|
|
d = d * 2;
|
||
|
|
else if ((a < 5000 && b > 20000) || (b < 5000 && a > 20000))
|
||
|
|
d = d * 3 / 2;
|
||
|
|
|
||
|
|
return d;
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifndef EXPLAIN
|
||
|
|
static uint image_signatures_dist_fuzzy(struct image_signature *sig1, struct image_signature *sig2)
|
||
|
|
#else
|
||
|
|
static uint image_signatures_dist_fuzzy_explain(struct image_signature *sig1, struct image_signature *sig2, void (*msg)(byte *text, void *param), void *param)
|
||
|
|
#endif
|
||
|
|
{
|
||
|
|
#ifdef EXPLAIN
|
||
|
|
byte buf[1024], *line = buf;
|
||
|
|
MSGL("Fuzzy matching");
|
||
|
|
explain_signature(sig1, msg, param);
|
||
|
|
explain_signature(sig2, msg, param);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
/* FIXME: do not mux textured and non-textured images (should be split in clusters tree) */
|
||
|
|
if ((sig1->flags ^ sig2->flags) & IMAGE_SIG_TEXTURED)
|
||
|
|
{
|
||
|
|
MSGL("Textured vs non-textured");
|
||
|
|
return ~0U;
|
||
|
|
}
|
||
|
|
|
||
|
|
uint cnt1 = sig1->len;
|
||
|
|
uint cnt2 = sig2->len;
|
||
|
|
struct image_region *reg1 = sig1->reg;
|
||
|
|
struct image_region *reg2 = sig2->reg;
|
||
|
|
uint mf[IMAGE_REG_MAX][IMAGE_REG_MAX], mh[IMAGE_REG_MAX][IMAGE_REG_MAX];
|
||
|
|
uint lf[IMAGE_REG_MAX * 2], lh[IMAGE_REG_MAX * 2];
|
||
|
|
uint df = sig1->df + sig2->df, dh = sig1->dh + sig2->dh;
|
||
|
|
|
||
|
|
/* Compute distance matrix */
|
||
|
|
for (uint i = 0; i < cnt1; i++)
|
||
|
|
for (uint j = 0; j < cnt2; j++)
|
||
|
|
{
|
||
|
|
uint d = 0;
|
||
|
|
for (uint k = 0; k < IMAGE_VEC_F; k++)
|
||
|
|
{
|
||
|
|
int dif = reg1[i].f[k] - reg2[j].f[k];
|
||
|
|
d += image_sig_cmp_features_weights[k] * dif * dif;
|
||
|
|
}
|
||
|
|
mf[i][j] = d;
|
||
|
|
d = 0;
|
||
|
|
for (uint k = 0; k < IMAGE_REG_H; k++)
|
||
|
|
{
|
||
|
|
int dif = reg1[i].h[k] - reg2[j].h[k];
|
||
|
|
d += image_sig_cmp_features_weights[k + IMAGE_VEC_F] * dif * dif;
|
||
|
|
}
|
||
|
|
mh[i][j] = d;
|
||
|
|
}
|
||
|
|
|
||
|
|
uint lfs = 0, lhs = 0;
|
||
|
|
for (uint i = 0; i < cnt1; i++)
|
||
|
|
{
|
||
|
|
uint f = mf[i][0], h = mh[i][0];
|
||
|
|
for (uint j = 1; j < cnt2; j++)
|
||
|
|
{
|
||
|
|
f = MIN(f, mf[i][j]);
|
||
|
|
h = MIN(h, mh[i][j]);
|
||
|
|
}
|
||
|
|
lf[i] = (df * 0x10000) / (df + fast_sqrt_u32(f));
|
||
|
|
lh[i] = (dh * 0x10000) / (dh + fast_sqrt_u32(h));
|
||
|
|
lfs += lf[i] * (6 * reg1[i].wa + 2 * reg1[i].wb);
|
||
|
|
lhs += lh[i] * reg1[i].wa;
|
||
|
|
}
|
||
|
|
for (uint i = 0; i < cnt2; i++)
|
||
|
|
{
|
||
|
|
uint f = mf[0][i], h = mh[0][i];
|
||
|
|
for (uint j = 1; j < cnt1; j++)
|
||
|
|
{
|
||
|
|
f = MIN(f, mf[j][i]);
|
||
|
|
h = MIN(h, mh[j][i]);
|
||
|
|
}
|
||
|
|
lf[i + cnt1] = (df * 0x10000) / (df + fast_sqrt_u32(f));
|
||
|
|
lh[i + cnt1] = (dh * 0x10000) / (dh + fast_sqrt_u32(h));
|
||
|
|
lfs += lf[i] * (6 * reg2[i].wa + 2 * reg2[i].wb);
|
||
|
|
lhs += lh[i] * reg2[i].wa;
|
||
|
|
}
|
||
|
|
|
||
|
|
uint measure = lfs * 6 + lhs * 2 * 8;
|
||
|
|
|
||
|
|
#ifdef EXPLAIN
|
||
|
|
/* Display similarity vectors */
|
||
|
|
MSG("Lf=(");
|
||
|
|
for (uint i = 0; i < cnt1 + cnt2; i++)
|
||
|
|
{
|
||
|
|
if (i)
|
||
|
|
MSG(" ");
|
||
|
|
if (i == cnt1)
|
||
|
|
MSG("~ ");
|
||
|
|
MSG("%.4f", (double)lf[i] / 0x10000);
|
||
|
|
}
|
||
|
|
MSGL(")");
|
||
|
|
MSG("Lh=(");
|
||
|
|
for (uint i = 0; i < cnt1 + cnt2; i++)
|
||
|
|
{
|
||
|
|
if (i)
|
||
|
|
MSG(" ");
|
||
|
|
if (i == cnt1)
|
||
|
|
MSG("~ ");
|
||
|
|
MSG("%.4f", (double)lh[i] / 0x10000);
|
||
|
|
}
|
||
|
|
MSGL(")");
|
||
|
|
MSGL("Lfm=%.4f", lfs / (double)(1 << (3 + 8 + 16)));
|
||
|
|
MSGL("Lhm=%.4f", lhs / (double)(1 << (8 + 16)));
|
||
|
|
MSGL("measure=%.4f", measure / (double)(1 << (3 + 3 + 8 + 16)));
|
||
|
|
#endif
|
||
|
|
|
||
|
|
return (1 << (3 + 3 + 8 + 16)) - measure;
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifndef EXPLAIN
|
||
|
|
static uint image_signatures_dist_average(struct image_signature *sig1, struct image_signature *sig2)
|
||
|
|
#else
|
||
|
|
static uint image_signatures_dist_average_explain(struct image_signature *sig1, struct image_signature *sig2, void (*msg)(byte *text, void *param), void *param)
|
||
|
|
#endif
|
||
|
|
{
|
||
|
|
#ifdef EXPLAIN
|
||
|
|
byte buf[1024], *line = buf;
|
||
|
|
MSGL("Average matching");
|
||
|
|
#endif
|
||
|
|
|
||
|
|
uint dist = 0;
|
||
|
|
for (uint i = 0; i < IMAGE_VEC_F; i++)
|
||
|
|
{
|
||
|
|
uint d = image_sig_cmp_features_weights[0] * isqr((int)sig1->vec.f[i] - (int)sig2->vec.f[i]);
|
||
|
|
MSGL("feature %u: d=%u (%u %u)", i, d, sig1->vec.f[i], sig2->vec.f[i]);
|
||
|
|
dist += d;
|
||
|
|
}
|
||
|
|
|
||
|
|
MSGL("dist=%u", dist);
|
||
|
|
return dist;
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifndef EXPLAIN
|
||
|
|
#define CALL(x) image_signatures_dist_##x(sig1, sig2)
|
||
|
|
uint image_signatures_dist(struct image_signature *sig1, struct image_signature *sig2)
|
||
|
|
#else
|
||
|
|
#define CALL(x) image_signatures_dist_##x##_explain(sig1, sig2, msg, param)
|
||
|
|
uint image_signatures_dist_explain(struct image_signature *sig1, struct image_signature *sig2, void (*msg)(byte *text, void *param), void *param)
|
||
|
|
#endif
|
||
|
|
{
|
||
|
|
if (!sig1->len)
|
||
|
|
return CALL(average);
|
||
|
|
else
|
||
|
|
switch (image_sig_compare_method)
|
||
|
|
{
|
||
|
|
case 0:
|
||
|
|
return CALL(integrated);
|
||
|
|
case 1:
|
||
|
|
return CALL(fuzzy);
|
||
|
|
case 2:
|
||
|
|
return CALL(average);
|
||
|
|
default:
|
||
|
|
ASSERT(0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#undef CALL
|
||
|
|
|
||
|
|
#undef EXPLAIN
|
||
|
|
#undef MSG
|
||
|
|
#undef LINE
|
||
|
|
#undef MSGL
|