示例#1
0
    def __init__(self):
        super(RetrievalSystem, self).__init__()
        self.setWindowTitle('实验三-企业检索系统')

        self.resize(1280, 720)

        table1 = QTableWidget(self)
        table1.setColumnCount(3)
        table1.setHorizontalHeaderLabels(page_header)

        table2 = QTableWidget(self)
        table2.setColumnCount(3)
        table2.setHorizontalHeaderLabels(file_header)

        search = Search()
        self.tab1 = Retrieval(table1, search, 'page')
        self.tab2 = Retrieval(table2, search, 'file')
        self.addTab(self.tab1, "页面检索")
        self.addTab(self.tab2, "文档检索")
        self.show()
示例#2
0
import os
import io
import tempfile
import shutil
from PIL import Image
from werkzeug.utils import secure_filename
from flask import Flask, render_template, url_for, request, redirect, Markup
from utils import compactCode, Search

search_model = compactCode(centers_path=os.path.join("centers", "centers.h5py"),
                           pq_centers_path=os.path.join("centers", "pq_centers.h5py"),
                           codes_path=os.path.join("centers", "codes"),
                           codes_name=os.path.join("centers", "codes_name"))

model = Search(search_model)

# create temp folder
os.makedirs(os.path.join("static", "temp"), exist_ok=True)
# result
res_img = [os.path.join("temp", "result{:02}.jpg".format(i)) for i in range(10)]
# result combined
com_img = [os.path.join("temp", "combined{:02}.jpg".format(i)) for i in range(10)]

app = Flask(__name__)

ALLOWED_EXTENSIONS = {'jpg', 'jpeg', 'png'}
app.config['SEND_FILE_MAX_AGE_DEFAULT'] = 0


MAX_WIDTH = 800
MAX_HEIGHT = 550