Esempio n. 1
0
#

from http.server import HTTPServer, SimpleHTTPRequestHandler
import json
import logging
import time

import ikalog.constants
from ikalog.utils import *
from ikalog.utils.character_recoginizer import DeadlyWeaponRecoginizer
from ikalog.utils.neuralnet.weapon import WeaponClassifier
import cv2
import numpy as np
import umsgpack

weapons = WeaponClassifier()
weapons.load_model_from_file()

abilities = GearPowerRecoginizer()
abilities.load_model_from_file()
abilities.knn_train()


class APIServer(object):
    def _decode_deadly_weapons_image(self, payload):
        h = payload['sample_height']
        w = payload['sample_width']
        img_bytes = payload['samples']
        samples1 = cv2.imdecode(np.fromstring(img_bytes, dtype='uint8'), 1)
        num_samples = int(samples1.shape[0] / h)