1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30 package org.openimaj.image.feature.dense.gradient.dsift;
31
32 import java.io.DataInput;
33 import java.io.DataOutput;
34 import java.io.IOException;
35 import java.io.PrintWriter;
36 import java.util.Scanner;
37 import java.util.StringTokenizer;
38
39 import org.openimaj.feature.ByteFV;
40
41
42
43
44
45
46
47
48
49 public class ByteDSIFTKeypoint
50 extends
51 AbstractDSIFTKeypoint<ByteFV, byte[]>
52 {
53 static final long serialVersionUID = 12345545L;
54
55
56
57
58 public ByteDSIFTKeypoint() {
59 this(DEFAULT_LENGTH);
60 }
61
62
63
64
65
66
67
68 public ByteDSIFTKeypoint(final int length) {
69 this.descriptor = new byte[length];
70 }
71
72
73
74
75
76
77
78
79
80
81
82
83
84 public ByteDSIFTKeypoint(final float x, final float y, final byte[] descriptor, final float energy) {
85 this.x = x;
86 this.y = y;
87 this.descriptor = descriptor;
88 this.energy = energy;
89 }
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105 public ByteDSIFTKeypoint(final float x, final float y, final float[] fdescriptor, final float energy) {
106 this.x = x;
107 this.y = y;
108 this.energy = energy;
109 this.descriptor = new byte[fdescriptor.length];
110
111 for (int i = 0; i < descriptor.length; i++) {
112 final int intval = (int) (512.0 * fdescriptor[i]);
113 descriptor[i] = (byte) (Math.min(255, intval) - 128);
114 }
115 }
116
117 @Override
118 public ByteFV getFeatureVector() {
119 return new ByteFV(descriptor);
120 }
121
122 @Override
123 public String toString() {
124 return ("ByteDSIFTKeypoint(" + this.x + ", " + this.y + ")");
125 }
126
127 @Override
128 public void writeBinary(DataOutput out) throws IOException {
129 out.writeFloat(x);
130 out.writeFloat(y);
131 out.writeFloat(energy);
132
133 out.write(this.descriptor);
134 }
135
136 @Override
137 public void writeASCII(PrintWriter out) throws IOException {
138
139 out.write(x + " " + y + " " + energy + "\n");
140
141 for (int i = 0; i < descriptor.length; i++) {
142 if (i > 0 && i % 20 == 0)
143 out.println();
144 out.print(" " + (descriptor[i] + 128));
145 }
146 out.println();
147 }
148
149 @Override
150 public void readBinary(DataInput in) throws IOException {
151 x = in.readFloat();
152 y = in.readFloat();
153 energy = in.readFloat();
154 in.readFully(descriptor);
155 }
156
157 @Override
158 public void readASCII(Scanner in) throws IOException {
159 x = in.nextFloat();
160 y = in.nextFloat();
161 energy = in.nextFloat();
162
163 int i = 0;
164 while (i < descriptor.length) {
165 final String line = in.nextLine();
166 final StringTokenizer st = new StringTokenizer(line);
167
168 while (st.hasMoreTokens()) {
169 descriptor[i] = (byte) (Integer.parseInt(st.nextToken()) - 128);
170 i++;
171 }
172 }
173 }
174 }