[1044] | 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 |
---|

[1063] | 15 | E-Mail: rubner@cs.stanford.edu URL: http://robotics.stanford.edu/~rubner/emd/default.htm |
---|

[1044] | 16 | */ |
---|

| 17 | |
---|

| 18 | /* DEFINITIONS */ |
---|

| 19 | #define MAX_ITERATIONS 500 |
---|

| 20 | //#define INFINITY 1e20 |
---|

| 21 | #define EPSILON 1e-6 |
---|

| 22 | |
---|

| 23 | /*****************************************************************************/ |
---|

| 24 | /* feature_t SHOULD BE MODIFIED BY THE USER TO REFLECT THE FEATURE TYPE */ |
---|

| 25 | //typedef struct { |
---|

| 26 | // double X,Y,Z; |
---|

| 27 | //} |
---|

| 28 | typedef double feature_t; |
---|

| 29 | /*****************************************************************************/ |
---|

| 30 | |
---|

| 31 | |
---|

| 32 | typedef struct |
---|

| 33 | { |
---|

| 34 | int n; /* Number of features in the signature */ |
---|

| 35 | feature_t *Features; /* Pointer to the features vector */ |
---|

| 36 | float *Weights; /* Pointer to the weights of the features */ |
---|

| 37 | } signature_t; |
---|

| 38 | |
---|

| 39 | |
---|

| 40 | typedef struct |
---|

| 41 | { |
---|

| 42 | int from; /* Feature number in signature 1 */ |
---|

| 43 | int to; /* Feature number in signature 2 */ |
---|

| 44 | float amount; /* Amount of flow from "from" to "to" */ |
---|

| 45 | } flow_t; |
---|

| 46 | |
---|

| 47 | |
---|

| 48 | |
---|

| 49 | float emd(signature_t *Signature1, signature_t *Signature2, |
---|

| 50 | float (*func)(feature_t *, feature_t *), |
---|

| 51 | flow_t *Flow, int *FlowSize); |
---|

| 52 | |
---|

| 53 | #endif |
---|