def Main():
try:
# Get application arguments
ap = ArgumentParser(ArgumentsType.SERVER)
# Get port number
grpcServerPort = ap.GetGrpcServerPort()
logger.info('gRPC server port: {0}'.format(grpcServerPort))
# init graph topology & instance
counter = 0
while init_topology() == -1:
if counter == 100:
logger.critical('Failed to init topology, please check whether direct method still works')
exit(-1)
logger.warning('Failed to init topology, try again 10 secs later')
time.sleep(10)
counter += 1
# create gRPC server and start running
server = grpc.server(futures.ThreadPoolExecutor(max_workers=3))
extension_pb2_grpc.add_MediaGraphExtensionServicer_to_server(
InferenceEngine(stream_manager), server)
server.add_insecure_port(f'[::]:{grpcServerPort}')
server.start()
app.run(host='0.0.0.0', debug=False)
# server.wait_for_termination()
except:
PrintGetExceptionDetails()
raise
def main():
"""main.
Main loop.
"""
try:
# Get application arguments
argument_parser = ArgumentParser(ArgumentsType.SERVER)
if IS_OPENCV == "false":
# Get port number
grpcServerPort = argument_parser.GetGrpcServerPort()
logger.info("gRPC server port: %s", grpcServerPort)
# init graph topology & instance
counter = 0
while init_topology() == -1:
if counter == 100:
logger.critical(
"Failed to init topology, please check whether direct method still works"
)
sys.exit(-1)
logger.warning(
"Failed to init topology, try again 10 secs later")
time.sleep(10)
counter += 1
# create gRPC server and start running
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
extension_pb2_grpc.add_MediaGraphExtensionServicer_to_server(
InferenceEngine(stream_manager), server
)
server.add_insecure_port(f"[::]:{grpcServerPort}")
server.start()
else:
logger.info("opencv server")
# opencv_zmq()
uvicorn.run(app, host="0.0.0.0", port=5000)
# server.wait_for_termination()
except:
PrintGetExceptionDetails()
raise
def Main():
try:
# Get application arguments
ap = ArgumentParser(ArgumentsType.SERVER)
# Get port number
grpcServerPort = ap.GetGrpcServerPort()
logging.info('gRPC server port: {0}'.format(grpcServerPort))
# create gRPC server and start running
server = grpc.server(futures.ThreadPoolExecutor(max_workers=3))
extension_pb2_grpc.add_MediaGraphExtensionServicer_to_server(
InferenceEngine(), server)
server.add_insecure_port(f'[::]:{grpcServerPort}')
server.start()
server.wait_for_termination()
except:
PrintGetExceptionDetails()
exit(-1)
def parse_args(args=sys.argv[1:]):
from arguments import (ArgumentParser, ParserGroup, TFArgparser,
CSVGeneratorArgparser, NMSArgparser,
InferenceArgparer)
parser = ArgumentParser()
parser.add_argument('--annotate-thresh', type=float, default=0.9)
parser.add_argument('--thresh-step', type=float, default=0.05)
parser.add_argument('--name')
return ParserGroup(TFArgparser(),
CSVGeneratorArgparser(data_train=False),
NMSArgparser(),
InferenceArgparer(),
parser,
name='transfer').parse(args)
for i in range(outputs.size(0)):
pred_label_img = outputs.cpu().numpy()[i]
pred_label_img = pred_label_img - 1
pred_label_img.astype(np.uint8)
img_id = imgPT[i]
pred_label_img = trainId_to_id_map_func(
pred_label_img) # (shape: (1024, 2048))
pred_label_img = pred_label_img.astype(np.uint8)
cv2.imwrite(
args.save_dir + "/" + img_id + "_pred_label_img.png",
pred_label_img)
if __name__ == '__main__':
visulparser = ArgumentParser()
# Execution mode
visulparser.add_argument("--model", default='ENet')
visulparser.add_argument(
"--model_path",
default='/media/tan/Samsung_T5/1024x2048/ENet_vanilla.pth')
# Hyperparameters
visulparser.add_argument("--dataset",
choices=['Cityscapes'],
default='Cityscapes')
visulparser.add_argument("--dataset_dir", type=str, default="./Cityscapes")