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 18:05:18 +08:00
parent 9ed6e4ef13
commit eca100066d
3 changed files with 21 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ieemoo-ai-zhanting.py
View File

@ -20,8 +20,8 @@ app = Flask(__name__)
app.use_reloader=False
parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default='runs/zhanting/yolov5l_finetune/exp/weights/best.pt', help='model.pt path(s)')
parser.add_argument('--source', type=str, default='data/images/1.jpg', help='source') # file/folder, 0 for webcam
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)')
parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')