16#ifndef TESSERACT_TRAINING_TRAININGSAMPLESET_H_
17#define TESSERACT_TRAINING_TRAININGSAMPLESET_H_
35struct UnicharAndFonts;
56 return samples_.size();
59 return num_raw_samples_;
68 return unicharset_size_;
71 return fontinfo_table_;
125 int font_id2,
int class_id2,
130 int font_id2,
int class_id2,
144 int font_id2,
int class_id2,
146 bool thorough)
const;
162 return samples_[index];
167 samples_[index] =
nullptr;
225 struct FontClassDistance {
231 struct FontClassInfo {
235 bool Serialize(FILE* fp)
const;
238 bool DeSerialize(
bool swap, FILE* fp);
241 int32_t num_raw_samples;
243 int32_t canonical_sample;
245 float canonical_dist;
253 BitVector cloud_features;
266 PointerVector<TrainingSample> samples_;
268 int num_raw_samples_;
272 int unicharset_size_;
276 IndexMapBiDi font_id_map_;
283 const FontInfoTable& fontinfo_table_;
int SparseSize() const override
STRING SampleToString(const TrainingSample &sample) const
const TrainingSample * GetCanonicalSample(int font_id, int class_id) const
int NumClassSamples(int font_id, int class_id, bool randomize) const
int AddSample(const char *unichar, TrainingSample *sample)
TrainingSample * extract_sample(int index)
const GenericVector< int > & GetCanonicalFeatures(int font_id, int class_id) const
bool Serialize(FILE *fp) const
const FontInfoTable & fontinfo_table() const
void AddAllFontsForClass(int class_id, Shape *shape) const
void IndexFeatures(const IntFeatureSpace &feature_space)
float ComputeClusterDistance(int font_id1, int class_id1, int font_id2, int class_id2, const IntFeatureMap &feature_map) const
void KillSample(TrainingSample *sample)
void ComputeCanonicalSamples(const IntFeatureMap &map, bool debug)
void ComputeCloudFeatures(int feature_space_size)
bool DeSerialize(bool swap, FILE *fp)
int num_raw_samples() const
void LoadUnicharset(const char *filename)
const BitVector & GetCloudFeatures(int font_id, int class_id) const
const UNICHARSET & unicharset() const
int GlobalSampleIndex(int font_id, int class_id, int index) const
void ComputeCanonicalFeatures()
TrainingSample * mutable_sample(int index)
void OrganizeByFontAndClass()
bool DeleteableSample(const TrainingSample *sample)
float ClusterDistance(int font_id1, int class_id1, int font_id2, int class_id2, const IntFeatureMap &feature_map)
float GetCanonicalDist(int font_id, int class_id) const
void ReplicateAndRandomizeSamples()
const TrainingSample * GetSample(int index) const
void DisplaySamplesWithFeature(int f_index, const Shape &shape, const IntFeatureSpace &feature_space, ScrollView::Color color, ScrollView *window) const
int ReliablySeparable(int font_id1, int class_id1, int font_id2, int class_id2, const IntFeatureMap &feature_map, bool thorough) const
TrainingSample * MutableSample(int font_id, int class_id, int index)
float UnicharDistance(const UnicharAndFonts &uf1, const UnicharAndFonts &uf2, bool matched_fonts, const IntFeatureMap &feature_map)