lichen/ieemoo-ai-zhanting
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
lichen
2022-04-11 20:28:15 +08:00
parent cbeeed4c4e
commit c1d82fafc4
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ieemoo-ai-zhanting.py
View File

@ -20,7 +20,7 @@ app = Flask(__name__)
app.use_reloader=False app.use_reloader=False
parser = argparse.ArgumentParser() 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('--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('--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)') parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
@ -58,6 +58,7 @@ def get_isempty():
} }
try: try:
imgdata = base64.b64decode(pic) imgdata = base64.b64decode(pic)
#image_path = 'data/images/1.jpg'
image_path = '../module/ieemoo-ai-zhanting/imgs/1.jpg' image_path = '../module/ieemoo-ai-zhanting/imgs/1.jpg'
file = open(image_path, 'wb') file = open(image_path, 'wb')
file.write(imgdata) file.write(imgdata)
@ -76,7 +77,7 @@ def get_isempty():
if __name__ == "__main__": if __name__ == "__main__":
app.run() app.run()
#http_server = WSGIServer(('0.0.0.0', 8082), app) #http_server = WSGIServer(('0.0.0.0', 8000), app)
#http_server.serve_forever() #http_server.serve_forever()