• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• stn-ocr-master
  • datasets
    • text_recognition
      • ctc_char_map.json
    • svhn
      • svhn_dataextract_tojson.py
      • create_svhn_dataset_4_images.py
      • create_svhn_dataset.py
      • README.md
      • filter_large_images.py
      • requirements.txt
      • svhn_char_map.json
    • fsns
      • slice_fsns_dataset.py
      • fsns_char_map.json
      • download_fsns.py
      • swap_classes.py
      • fonts
      • __init__.py
      • render_text_on_signs.py
      • transform_gt.py
      • tfrecord_utils
        • tfrecord_to_image.py
  • mxnet
    • callbacks
      • fsns_bbox_plotter.py
      • create_checkpoint.py
      • __init__.py
      • save_bboxes.py
    • utils
      • create_record_io_file.py
      • extract_last_image_from_gif.py
      • show_progress.py
      • bbox_utils.py
      • create_video.py
      • reorder_lst_file.py
      • plot_log.py
      • create_text_rec_gt.py
      • __init__.py
      • datatypes.py
      • extract_images.py
      • create_gif.py
    • train_fsns.py
    • train_utils.py
    • train_text_recognition.py
    • initializers
      • spn_initializer.py
      • __init__.py
    • eval_fsns_model.py
    • eval_text_recognition_model.py
    • operations
      • disable_shearing.py
      • ones.py
      • __init__.py
      • debug.py
    • networks
      • fsns.py
      • svhn.py
      • __init__.py
      • text_rec.py
    • eval_svhn_model.py
    • data_io
      • fsns_file_iter.py
      • lstm_iter.py
      • __init__.py
      • file_iter.py
    • metrics
      • fsns_metrics.py
      • test_ctc_loss.py
      • ctc_metrics.py
      • ctc
        • setup.py
        • ctc_loss.pyx
        • __init__.py
        • cctc_loss.pxd
      • __init__.py
      • base.py
    • symbols
      • lstm.py
      • __init__.py
    • train_svhn.py
  • LICENSE.md
  • README.md
  • requirements.txt
  • .gitignore
  • docs
    • images

import argparse
import json
import threading
import tkinter
from socketserver import BaseRequestHandler, ThreadingUDPServer, DatagramRequestHandler, ThreadingMixIn, UDPServer, \
    StreamRequestHandler, TCPServer

import base64
import numpy as np
from PIL import Image, ImageTk


class ProgressWindow:

    def __init__(self, root):
        self.frame = tkinter.Frame(root)
        self.frame.pack(fill=tkinter.BOTH, expand=tkinter.YES)

        self._image = None
        self._sprite = None
        self.canvas = tkinter.Canvas(
            self.frame,
            width=850,
            height=400
        )
        self.canvas.pack(fill=tkinter.BOTH, expand=tkinter.YES)

    @property
    def image(self):
        return self._image

    @image.setter
    def image(self, value):
        image = ImageTk.PhotoImage(value)
        self._image = image
        self._sprite = self.canvas.create_image(value.width // 2, value.height // 2, image=self._image)
        self.canvas.config(width=value.width, height=value.height)


class ImageDataHandler(StreamRequestHandler):

    def __init__(self, *args, **kwargs):
        self.window = kwargs.pop('window')
        super(ImageDataHandler, self).__init__(*args, **kwargs)

    def handle(self):
        data = self.rfile.read()
        data = json.loads(data.decode('utf-8'))
        width = data['width']
        height = data['height']
        data = np.fromstring(base64.b64decode(data['image']), dtype=np.uint8)
        data = np.resize(data, (height, width, 3))
        image = Image.fromarray(data, mode='RGB')
        self.window.image = image


class ImageServer(ThreadingMixIn, TCPServer):

    def __init__(self, *args, **kwargs):
        self.window = kwargs.pop('window')
        super(ImageServer, self).__init__(*args, **kwargs)

    def finish_request(self, request, client_address):
        self.RequestHandlerClass(request, client_address, self, window=self.window)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Tool that shows current pictures of a training')
    parser.add_argument('--host', default='0.0.0.0', help='address to listen on')
    parser.add_argument('--port', type=int, default=1337, help='port to listen on')

    args = parser.parse_args()

    root = tkinter.Tk()
    window = ProgressWindow(root)

    print("starting server")
    server = ImageServer((args.host, args.port), ImageDataHandler, window=window)
    server_thread = threading.Thread(target=server.serve_forever)
    server_thread.deamon = True
    server_thread.start()

    print("starting window")
    root.mainloop()
    server.shutdown()
    server.server_close()