32 lines
848 B
Python
32 lines
848 B
Python
import logging
|
|
from multiprocessing import Process, Queue
|
|
from trajpred.config import parser
|
|
from trajpred.prediction_server import InferenceServer, run_inference_server
|
|
from trajpred.socket_forwarder import run_ws_forwarder
|
|
|
|
|
|
logger = logging.getLogger("trajpred.plumbing")
|
|
|
|
def start():
|
|
args = parser.parse_args()
|
|
loglevel = logging.NOTSET if args.verbose > 1 else logging.DEBUG if args.verbose > 0 else logging.INFO
|
|
logging.basicConfig(
|
|
level=loglevel,
|
|
)
|
|
|
|
# instantiating process with arguments
|
|
procs = [
|
|
Process(target=run_ws_forwarder, args=(args,))
|
|
]
|
|
if not args.bypass_prediction:
|
|
procs.append(
|
|
Process(target=run_inference_server, args=(args,)),
|
|
)
|
|
|
|
logger.info("start")
|
|
for proc in procs:
|
|
proc.start()
|
|
|
|
for proc in procs:
|
|
proc.join()
|
|
|