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

update

Browse Source
This commit is contained in:
lichen
2022-05-05 13:54:16 +08:00
parent cbcdb2bf8a
commit a79ed4db47
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ieemoo-ai-predict.py
View File

@ -37,7 +37,7 @@ app = Flask(__name__)
load model
'''
parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-predict/', help='model.pt path(s)')
parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-predict/model/now/', help='model.pt path(s)')
parser.add_argument('--Form', type=str, default='')
parser.add_argument('--source', type=str, default='../module/ieemoo-ai-predict/imgs', help='source') # file/folder, 0 for webcam
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')