source: cpp/frams/model/similarity/EMD/emd.h @ 1044

Last change on this file since 1044 was 1044, checked in by oriona, 3 years ago

Similarity measures code refactored. Distribution-based similarity measure added.

File size: 1.5 KB
Line 
1#ifndef _EMD_H
2#define _EMD_H
3/*
4    emd.h
5
6    Last update: 3/24/98
7
8    An implementation of the Earth Movers Distance.
9    Based of the solution for the Transportation problem as described in
10    "Introduction to Mathematical Programming" by F. S. Hillier and
11    G. J. Lieberman, McGraw-Hill, 1990.
12
13    Copyright (C) 1998 Yossi Rubner
14    Computer Science Department, Stanford University
15    E-Mail: rubner@cs.stanford.edu   URL: http://vision.stanford.edu/~rubner
16*/
17
18/* DEFINITIONS */
19#define MAX_SIG_SIZE   1000
20#define MAX_ITERATIONS 500
21//#define INFINITY       1e20
22#define EPSILON        1e-6
23
24/*****************************************************************************/
25/* feature_t SHOULD BE MODIFIED BY THE USER TO REFLECT THE FEATURE TYPE      */
26//typedef struct {
27//   double X,Y,Z;
28//}
29typedef double feature_t;
30/*****************************************************************************/
31
32
33typedef struct
34{
35  int n;                /* Number of features in the signature */
36  feature_t *Features;  /* Pointer to the features vector */
37  float *Weights;       /* Pointer to the weights of the features */
38} signature_t;
39
40
41typedef struct
42{
43  int from;             /* Feature number in signature 1 */
44  int to;               /* Feature number in signature 2 */
45  float amount;         /* Amount of flow from "from" to "to" */
46} flow_t;
47
48
49
50float emd(signature_t *Signature1, signature_t *Signature2,
51          float (*func)(feature_t *, feature_t *),
52          flow_t *Flow, int *FlowSize);
53
54#endif
Note: See TracBrowser for help on using the repository browser.