From c1d82fafc4f63b33aefc9fc4e32165715add5e12 Mon Sep 17 00:00:00 2001 From: lichen <770918727@qq.com> Date: Mon, 11 Apr 2022 20:28:15 +0800 Subject: [PATCH] update --- ieemoo-ai-zhanting.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ieemoo-ai-zhanting.py b/ieemoo-ai-zhanting.py index c61e6d8..a1c70e3 100755 --- a/ieemoo-ai-zhanting.py +++ b/ieemoo-ai-zhanting.py @@ -20,7 +20,7 @@ app = Flask(__name__) app.use_reloader=False parser = argparse.ArgumentParser() -#parser.add_argument('--weights', nargs='+', type=str, default='runs/zhanting/yolov5l/best_1_30.pt', help='model.pt path(s)') +#parser.add_argument('--weights', nargs='+', type=str, default='runs/zhanting/yolov5m_finetune/exp4/weights/best.pt', help='model.pt path(s)') parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-zhanting/best.pt', help='model.pt path(s)') parser.add_argument('--source', type=str, default='../module/ieemoo-ai-zhanting/imgs/1.jpg', help='source') # file/folder, 0 for webcam parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)') @@ -58,6 +58,7 @@ def get_isempty(): } try: imgdata = base64.b64decode(pic) + #image_path = 'data/images/1.jpg' image_path = '../module/ieemoo-ai-zhanting/imgs/1.jpg' file = open(image_path, 'wb') file.write(imgdata) @@ -76,7 +77,7 @@ def get_isempty(): if __name__ == "__main__": app.run() - #http_server = WSGIServer(('0.0.0.0', 8082), app) + #http_server = WSGIServer(('0.0.0.0', 8000), app) #http_server.serve_forever()