Skip to content

Commit

Permalink
layout analysis: option for 'global' orientation estimation added & t…
Browse files Browse the repository at this point in the history
…abe cell conversion fixed
  • Loading branch information
diemmarkus committed Apr 20, 2017
1 parent e78cbfc commit 98154eb
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 2 deletions.
13 changes: 11 additions & 2 deletions Modules/Layout/src/LayoutPlugin.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -104,8 +104,12 @@ LayoutPlugin::LayoutPlugin(QObject* parent) : QObject(parent) {
mConfig.saveDefaultSettings(s);
rdf::SuperPixelTrainerConfig spc;
spc.saveDefaultSettings(s);

rdf::SuperPixelLabelerConfig splc;
splc.saveDefaultSettings(s);

rdf::LayoutAnalysisConfig lac;
lac.saveDefaultSettings(s);

s.endGroup();
}
Expand Down Expand Up @@ -158,6 +162,7 @@ void LayoutPlugin::saveSettings(QSettings & settings) const {
mConfig.saveSettings(settings);
mSplConfig.saveSettings(settings);
mSpcConfig.saveSettings(settings);
mLAConfig.saveSettings(settings);
//mLTRConfig.saveSettings(settings);
settings.endGroup();
}
Expand All @@ -169,6 +174,7 @@ void LayoutPlugin::loadSettings(QSettings & settings) {
mConfig.loadSettings(settings);
mSplConfig.loadSettings(settings);
mSpcConfig.loadSettings(settings);
mLAConfig.loadSettings(settings);
//mLTRConfig.loadSettings(settings);
settings.endGroup();
}
Expand Down Expand Up @@ -329,6 +335,7 @@ cv::Mat LayoutPlugin::compute(const cv::Mat & src, rdf::PageXmlParser & parser)

// compute layout analysis
rdf::LayoutAnalysis la(img);
la.setConfig(QSharedPointer<rdf::LayoutAnalysisConfig>(new rdf::LayoutAnalysisConfig(mLAConfig)));
la.setRootRegion(pe->rootRegion());

if (!la.compute())
Expand All @@ -341,7 +348,9 @@ cv::Mat LayoutPlugin::compute(const cv::Mat & src, rdf::PageXmlParser & parser)

auto root = la.textBlockSet().toTextRegion();
for (const QSharedPointer<rdf::Region>& r : root->children()) {
pe->rootRegion()->addUniqueChild(r, true); // true -> update

if (!pe->rootRegion()->reassignChild(r))
pe->rootRegion()->addUniqueChild(r, true); // true -> update
}

qInfo() << "layout analysis computed in" << dt;
Expand Down
2 changes: 2 additions & 0 deletions Modules/Layout/src/LayoutPlugin.h
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@ related links:
#include "LineTrace.h"
#include "SuperPixelTrainer.h"
#include "SuperPixelClassification.h"
#include "LayoutAnalysis.h"

// opencv defines
namespace cv {
Expand Down Expand Up @@ -131,6 +132,7 @@ class LayoutPlugin : public QObject, nmc::DkBatchPluginInterface {
rdf::LineTraceConfig mLTRConfig;
rdf::SuperPixelLabelerConfig mSplConfig;
rdf::SuperPixelClassifierConfig mSpcConfig;
rdf::LayoutAnalysisConfig mLAConfig;
LayoutConfig mConfig;

// layout plugin functions
Expand Down

0 comments on commit 98154eb

Please sign in to comment.