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
}
final int[][] planeVectors = plane.toQuantizationVectors(options.getVectorDimension());
Log("PlaneVectorCount: %d", planeVectors.length);
if (!hasGeneralQuantizer) {
Log(String.format("Training vector quantizer from plane %d.", planeIndex));
......
......@@ -693,14 +693,27 @@ public class LBGVectorQuantizer {
* @return Array of training vectors.
*/
private TrainingVector[] getEntryTrainingVectors(final int entryIndex, final int vectorCount) {
TrainingVector[] vectors = new TrainingVector[vectorCount];
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) {
if (trainingVector.getEntryIndex() == entryIndex) {
vectors[index++] = trainingVector;
}
}
assert (index == vectorCount);
return vectors;
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment