def __init__(self, parent=None):
super().__init__(parent)
self.ui = extract_ctrl_ui.Ui_ExtractCtrlWidget()
self.ui.setupUi(self)
self.setWindowIcon(QIcon(':/images/icon.png'))
self.setAttribute(Qt.WA_QuitOnClose, False)
extract_conf = TdConfig(
AppSettings.config_file_path).getExtractConfig()
self.ui.spinbox_mser_delta.setValue(
extract_conf[TdExtractConfigKey.DELTA])
self.ui.spinbox_mser_area_high.setValue(
extract_conf[TdExtractConfigKey.AREA_MAX])
self.ui.spinbox_mser_area_low.setValue(
extract_conf[TdExtractConfigKey.AREA_MIN])
self.ui.dspinbox_mser_variation.setValue(
extract_conf[TdExtractConfigKey.VARIATION])
for src in extract_conf[TdExtractConfigKey.SRCS]:
self.ui.listWidget.addItem(QListWidgetItem(src))
filter_conf = TdConfig(
AppSettings.config_file_path).getFilterConfig('extract')
self.ui.spinbox_filter_area_lim.setValue(
filter_conf[TdFilterConfigKey.AREA_LIM])
self.ui.spinbox_filter_perimeter_low.setValue(
filter_conf[TdFilterConfigKey.PERIMETER_LIM][0])
self.ui.spinbox_filter_perimeter_high.setValue(
filter_conf[TdFilterConfigKey.PERIMETER_LIM][1])
self.ui.dspinbox_filter_aspect_ratio_low.setValue(
filter_conf[TdFilterConfigKey.ASPECT_RATIO_LIM][0])
self.ui.dspinbox_filter_aspect_ratio_high.setValue(
filter_conf[TdFilterConfigKey.ASPECT_RATIO_LIM][1])
self.ui.checkbox_filter_abs_aspect_ratio.setChecked(
filter_conf[TdFilterConfigKey.ASPECT_RATIO_GT1])
self.ui.dspinbox_filter_occupation_low.setValue(
filter_conf[TdFilterConfigKey.OCCUPATION_LIM][0])
self.ui.dspinbox_filter_occupation_high.setValue(
filter_conf[TdFilterConfigKey.OCCUPATION_LIM][1])
self.ui.dspinbox_filter_compactness_low.setValue(
filter_conf[TdFilterConfigKey.COMPACTNESS_LIM][0])
self.ui.dspinbox_filter_compactness_high.setValue(
filter_conf[TdFilterConfigKey.COMPACTNESS_LIM][1])
self.ui.spinbox_filter_width_low.setValue(
filter_conf[TdFilterConfigKey.WIDTH_LIM][0])
self.ui.spinbox_filter_width_high.setValue(
filter_conf[TdFilterConfigKey.WIDTH_LIM][1])
self.ui.spinbox_filter_height_low.setValue(
filter_conf[TdFilterConfigKey.HEIGHT_LIM][0])
self.ui.spinbox_filter_height_high.setValue(
filter_conf[TdFilterConfigKey.HEIGHT_LIM][1])
self.setEnabledForFilter(filter_conf[TdFilterConfigKey.FLAG])
self.initConnects()
return
def doPreprocess(self):
''' 进行预处理
'''
# 获取参数，进行预处理
svcconf = self.control_panel.getConfiguration() \
if self.control_panel is not None else \
TdConfig(AppSettings.config_file_path).getSVCConfig()
ret = self.svc.init(svcconf.get(TdSVCConfigKey.MCONF_PATH, None))
if not ret:
return False
# 获取输入数据
logger.info("SVC require datas.")
self.requireData.emit()
logger.info("SVC tell data was recevied")
# SVC 检测过滤
final_tl = []
for item in self.input_data[0]:
tl = []
name, bg_image, tlregions = item
for region in tlregions:
sample, _ = crop_rect(bg_image, region)
ret = self.svc.predict(sample)
if ret:
tl.append(region)
final_tl.append((name, tlregions, tl))
# 最终合并
ret_final_mtl, ret_final_stl = [], []
for item in final_tl:
name, mtl, stl = item
ret_final_mtl.extend(mtl)
ret_final_stl.extend(stl)
merger = TdMergingOverlap()
merger.setConfig(
TdConfig(AppSettings.config_file_path).getMergeTLConfig())
tlsin1 = merger.mergeTextLine(ret_final_stl)
self.output_data = (ret_final_mtl, ret_final_stl, tlsin1)
# 显示结果
rgb_image = self.input_data[1].copy()
rgb_image = TdMergingTextLine.drawRegions(rgb_image, (255, 255, 255),
cv2.LINE_4, ret_final_mtl)
rgb_image = TdMergingTextLine.drawRegions(rgb_image, (0, 255, 0),
cv2.LINE_4, tlsin1)
self.setDisplayCvImage(rgb_image)
return self.output_data
def getConfig(self):
''' 获取配置
'''
megconf = self.control_panel.getConfiguration() \
if self.control_panel is not None else \
TdConfig(AppSettings.config_file_path).getMergeTLConfig()
return megconf
def __init__(self, parent=None):
super().__init__(parent)
self.ui = merge_ctrl_ui.Ui_MergeCtrlWidget()
self.ui.setupUi(self)
self.setWindowIcon(QIcon(':/images/icon.png'))
self.setAttribute(Qt.WA_QuitOnClose, False)
# 填充 MergeTLE 参数
merge_conf = TdConfig(AppSettings.config_file_path).getMergeTLConfig()
self.ui.linedit_combined_area_size_lim.setText(
str(merge_conf[TdMergeTLConfigKey.COMBINED_AREA_SIZE_LIM]))
self.ui.dspinbox_overlap_ratio.setValue(
merge_conf[TdMergeTLConfigKey.OVERLAP_RATIO])
self.ui.dspinbox_distance.setValue(
merge_conf[TdMergeTLConfigKey.DISTANCE])
self.ui.dspinbox_combined_aspect_ratio_low.setValue(
merge_conf[TdMergeTLConfigKey.COMBINED_ASPECT_RATIO_LIM][0])
self.ui.dspinbox_combined_aspect_ratio_high.setValue(
merge_conf[TdMergeTLConfigKey.COMBINED_ASPECT_RATIO_LIM][1])
self.ui.combobox_strategy.setCurrentIndex(
merge_conf[TdMergeTLConfigKey.STRATEGY].value - 1)
self.ui.combobox_position_ratio.setCurrentIndex(
merge_conf[TdMergeTLConfigKey.POSITION_RATIO].value - 1)
self.ui.dspinbox_position_ratio.setValue(
merge_conf[TdMergeTLConfigKey.POSITION_RATIO_CONSTANT])
self.ui.dspinbox_position_ratio.setVisible(
bool(merge_conf[TdMergeTLConfigKey.POSITION_RATIO] is
MergingPositionRatio.CONSTANT))
self.ui.linedit_scope_lim.setText(
str(merge_conf[TdMergeTLConfigKey.SCOPE_LIM]))
fltr_conf = TdConfig(
AppSettings.config_file_path).getFilterConfig("merge")
self.ui.checkbox_tl_area_lim_enable.setChecked(
bool(TdFilterCheckType.AREA in fltr_conf[TdFilterConfigKey.FLAG]))
self.ui.spinbox_tl_area_lim.setValue(
fltr_conf[TdFilterConfigKey.AREA_LIM])
self.initConnects()
return
def doPreprocess(self):
''' 进行预处理
'''
# 获取参数,进行预处理
config = self.control_panel.getConfiguration() \
if self.control_panel is not None else \
TdConfig(AppSettings.config_file_path).getPrepConfig()
self.preprocesser.setConfig(config)
self.showVerbose(config)
# 显示预处理结果
self.cur_config = config
return
def __init__(self, parent=None):
super().__init__(parent)
self.ui = svc_ctrl_ui.Ui_SVCCtrlWidget()
self.ui.setupUi(self)
self.setWindowIcon(QIcon(':/images/icon.png'))
self.setAttribute(Qt.WA_QuitOnClose, False)
svc_conf = TdConfig(AppSettings.config_file_path).getSVCConfig()
self.mconf_path = svc_conf.get(TdSVCConfigKey.MCONF_PATH, None)
self.ui.linedit_svc_conf_path.setText(str(self.mconf_path))
if Path(self.mconf_path).exists():
with open(self.mconf_path, 'r') as f:
mconf = yaml.load(f, Loader=yaml.FullLoader)
self.ui.label_config_context.setText(str(mconf))
self.ui.btn_svc_conf_path.clicked.connect(self.onActionBtnClicked)
def onActionExtractorRequireData(self, srcs, extconf):
''' 获取 Extracter 所需的数据
Args:
srcs 源图像
extconf 配置信息
'''
# 设置预处理
config = TdConfig(AppSettings.config_file_path).getPrepConfig() \
if self.preprocess_display_widget.control_panel is None else \
self.preprocess_display_widget.control_panel.getConfiguration()
self.preprocess_display_widget.preprocesser.setConfig(config)
# 初始化配置
posi_extconf = extconf.copy()
nega_extconf = extconf.copy()
posi_extconf[TdExtractConfigKey.DIRECTION] = ExtractDirection.Positive
nega_extconf[TdExtractConfigKey.DIRECTION] = ExtractDirection.Negitive
# 准备预处理后图像
images, image = [], []
for src in srcs:
image.clear()
grayname, chartype = src.split('.')
if grayname == "Gray":
ret = self.preprocess_display_widget.preprocesser.ret_gray
elif grayname == "Red":
ret = self.preprocess_display_widget.preprocesser.ret_red
elif grayname == "Green":
ret = self.preprocess_display_widget.preprocesser.ret_green
elif grayname == "Blue":
ret = self.preprocess_display_widget.preprocesser.ret_blue
else:
ret = None
if chartype == "Both" or chartype == "Black":
image.append({'name':"%s.Black"%grayname,
'image':ret[1],
'conf':posi_extconf})
if chartype == "Both" or chartype == "Bright":
image.append({'name':"%s.Bright"%grayname,
'image':ret[2],
'conf':nega_extconf})
images.extend(image.copy())
self.extract_display_widget.input_images = images
msg = "Data is fed for extractor"
logger.info(msg)
def __init__(self):
self.image_path = None
self.image_name = None
self.config_file_path = None
self.gray_type = None
self.show_opts = []
self.makesample = False
self.save_option = CliSaveOptions.NONE
self.preprocessing = TdPreprocessing()
self.extracter = TdExtractConnectDomain()
self.filter = TdFilter()
self.merger = TdMergingTextLine()
self.svc = TdSVC()
self.ocr = TdOCR()
self.config = TdConfig()
self.morpher = TdMorphOperator()
def __init__(self, parent=None):
super().__init__(parent)
self.ui = prep_ctrl_ui.Ui_PrepCtrlWidget()
self.ui.setupUi(self)
self.setWindowIcon(QIcon(':/images/icon.png'))
self.setAttribute(Qt.WA_QuitOnClose, False)
prep_conf = TdConfig(AppSettings.config_file_path).getPrepConfig()
self.ui.linedit_total_pixels.setText(
str(prep_conf[TdPrepConfigKeys.TOTAL_PIXELS]))
self.ui.spinbox_bilateral_arg1.setValue(
prep_conf[TdPrepConfigKeys.BILATERAL][0])
self.ui.spinbox_bilateral_arg2.setValue(
prep_conf[TdPrepConfigKeys.BILATERAL][1])
self.ui.spinbox_bilateral_arg3.setValue(
prep_conf[TdPrepConfigKeys.BILATERAL][2])
self.ui.spinbox_gaussian_size.setValue(
prep_conf[TdPrepConfigKeys.GAUSS_SIZE])
self.ui.dspinbox_offset.setValue(prep_conf[TdPrepConfigKeys.OFFSET])
return
def doPreprocess(self):
''' 进行预处理
'''
# 获取并设置连通域提取配置参数
extconf = self.control_panel.getConfiguration(flag=0) \
if self.control_panel is not None else \
TdConfig(AppSettings.config_file_path).getExtractConfig()
fltconf = self.control_panel.getConfiguration(flag=1) \
if self.control_panel is not None else \
TdConfig(AppSettings.config_file_path).getFilterConfig("extract")
self.filter.setConfig(fltconf)
self.extracter.setConfig(extconf)
# 获取输入数据
msg = "Extractor require datas. Srcs:%s." % extconf[
TdExtractConfigKey.SRCS]
logger.info(msg)
self.requireData.emit(extconf[TdExtractConfigKey.SRCS], extconf)
logger.info("Extractor tell data was recevied")
# 提取连通域
binarizes, debug_data = [], []
self.extracter.debug.setEnable(
bool(extconf[TdExtractConfigKey.VERBOSE]))
self.filter.debug.setEnable(bool(extconf[TdExtractConfigKey.VERBOSE]))
for image in self.input_images:
binarized = self.extracter.extract_with_labels_for_images(
[image], self.filter)
binarizes.append((image['name'], binarized.copy(), image['image']))
if extconf[TdExtractConfigKey.
VERBOSE] and self.extracter.debug.enable:
debug_data.extend(self.extracter.debug.data)
# 显示 MSER 提取结果
if extconf[TdExtractConfigKey.VERBOSE] and self.extracter.debug.enable:
if self.dr_widget is None:
self.dr_widget = VerboseDisplayWidget()
self.dr_widget.setExtracterVerboseData(debug_data)
self.dr_widget.show()
# 获取并设置形态学处理配置参数
fltconf = self.control_panel.getConfiguration(flag=1) \
if self.control_panel is not None else \
TdConfig(AppSettings.config_file_path).getFilterConfig("morph")
self.filter.setConfig(fltconf)
# 形态学处理
self.output_regionlist = []
for binarized in binarizes:
regions = self.morpher.morph_operation(binarized[1], binarized[2],
self.filter)
self.output_regionlist.append(
((binarized[0], binarized[1], regions.copy())))
# 显示 MORPH 处理结果
if extconf[TdExtractConfigKey.VERBOSE_MORPH]:
verbose_data = {}
for item in self.output_regionlist:
verbose_data[item[0]] = self.morpher.getMaskImage(
item[1], item[2])
if self.dr_widget is None:
self.dr_widget = VerboseDisplayWidget()
self.dr_widget.setPrepVerboseData(verbose_data)
self.dr_widget.show()
return