Skip to content
Snippets Groups Projects
Commit 970c3098 authored by Vojtech Moravec's avatar Vojtech Moravec
Browse files

Added Hack to fix bug in getEntryTrainingVectors

parent 1cb49ba2
No related branches found
No related tags found
No related merge requests found
...@@ -131,7 +131,6 @@ public class VQImageCompressor extends CompressorDecompressorBase implements IIm ...@@ -131,7 +131,6 @@ public class VQImageCompressor extends CompressorDecompressorBase implements IIm
} }
final int[][] planeVectors = plane.toQuantizationVectors(options.getVectorDimension()); final int[][] planeVectors = plane.toQuantizationVectors(options.getVectorDimension());
Log("PlaneVectorCount: %d", planeVectors.length);
if (!hasGeneralQuantizer) { if (!hasGeneralQuantizer) {
Log(String.format("Training vector quantizer from plane %d.", planeIndex)); Log(String.format("Training vector quantizer from plane %d.", planeIndex));
......
...@@ -693,14 +693,27 @@ public class LBGVectorQuantizer { ...@@ -693,14 +693,27 @@ public class LBGVectorQuantizer {
* @return Array of training vectors. * @return Array of training vectors.
*/ */
private TrainingVector[] getEntryTrainingVectors(final int entryIndex, final int vectorCount) { private TrainingVector[] getEntryTrainingVectors(final int entryIndex, final int vectorCount) {
TrainingVector[] vectors = new TrainingVector[vectorCount];
int index = 0; int index = 0;
// TrainingVector[] vectors = new TrainingVector[vectorCount];
int count = 0;
for (final TrainingVector trainingVector : trainingVectors) {
if (trainingVector.getEntryIndex() == entryIndex) {
++count;
}
}
TrainingVector[] vectors = new TrainingVector[count];
// if (count > vectorCount) {
// System.err.println(String.format(
// "WARNING: got more training vectors than vectorCount! DIFF: %d",
// count - vectorCount));
// }
for (final TrainingVector trainingVector : trainingVectors) { for (final TrainingVector trainingVector : trainingVectors) {
if (trainingVector.getEntryIndex() == entryIndex) { if (trainingVector.getEntryIndex() == entryIndex) {
vectors[index++] = trainingVector; vectors[index++] = trainingVector;
} }
} }
assert (index == vectorCount);
return vectors; return vectors;
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment