diff --git a/ieemoo-ai-zhanting.py b/ieemoo-ai-zhanting.py old mode 100755 new mode 100644 index 629ab41..46eeb58 --- a/ieemoo-ai-zhanting.py +++ b/ieemoo-ai-zhanting.py @@ -40,7 +40,7 @@ app.use_reloader=False parser = argparse.ArgumentParser() #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/model/now/best.pt', help='model.pt path(s)') +parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-zhanting/model/now/best_cls23.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.60, help='object confidence threshold')