from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
import os, sys
if __name__ == '__main__':
sys.path.append(os.getcwd())
sys.path[0] = os.path.dirname(os.path.realpath(__file__))
from utils.configurator import Configurator
configurator = Configurator()
engine = create_engine(configurator.get('Database', 'connection_string'))
Session = sessionmaker(bind=engine)
Base = declarative_base()
from PIL import Image
import torch
import json
import numpy as np
import math
import os, sys
if __name__ == '__main__':
sys.path.append(os.getcwd())
sys.path[0] = os.path.dirname(os.path.realpath(__file__))
from utils.configurator import Configurator
configurator = Configurator()
DATASET_ROOT_PATH = configurator.get('Dataset', 'dataset_root_path')
CORRECT_DATASET_ROOT_PATH = configurator.get('Dataset',
'correct_dataset_root_path')
INCORRECT_DATASET_ROOT_PATH = configurator.get('Dataset',
'incorrect_dataset_root_path')
CORRECT_SAMPLES_NUMBER = 67048 # but 67049 officially declared
INCORRECT_SAMPLES_NUMBER = 66734
VGG_IMG_SIZE = 224
class LazyDataset(Dataset):
def __init__(self, mode='train'):
self._mode = mode
self._samples_path = []
import logging
from logging.handlers import RotatingFileHandler
if __name__ == '__main__':
sys.path.append(os.getcwd())
# comment out the following 2 lines to print debug info on stdout
logging.basicConfig(format='%(asctime)s %(message)s',
datefmt='%m/%d/%Y %I:%M:%S %p',
level=logging.INFO,
filename='maskednet.log')
logging.getLogger().addHandler(
RotatingFileHandler('maskednet.log', maxBytes=1024, backupCount=0))
sys.path[0] = os.path.dirname(os.path.realpath(__file__))
from server.srv import Server
from client.client import Client
from bridge.bridge import Bridge
from utils.configurator import Configurator
configurator = Configurator()
server = Server(address=configurator.get('Server', 'address'),
model_port=configurator.get('Server', 'model_port'),
bridge_port=configurator.get('Server', 'bridge_port'))
client = Client(server_uri=configurator.get('Client', 'server_uri'))
bridge = Bridge(server_uri=configurator.get('Bridge', 'server_uri'))
server.start()
client.start()
bridge.start()
