def slot_close_model(self):
"""
关闭模型槽函数
:return:
"""
self.pb_close_model.setDisabled(True)
predict_dl.close_sess()
self.pb_detect.setDisabled(True)
self.pb_load_model.setEnabled(True)
self.le_image_size.setEnabled(True)
def slot_load_model(self):
"""
加载模型槽函数
:return:
"""
# model = self.cb_model.currentText()
flag = predict_dl.load_model()
if flag == -1:
predict_dl.close_sess()
return
self.pb_load_model.setDisabled(True)
# self.pb_detect.setEnabled(True)
self.pb_close_model.setEnabled(True)