parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.pt', help='model.pt path(s)')
parser.add_argument('--source', type=str, default='data/video', help='source') # file/folder, 0 for webcam
weight设置Releases时,可以在官网上下载模型,放进项目文件夹,就能识别出来了。
source设置时,只需要精确到放东西的文件夹,不要包含图片、视频的名字。
parser.add_argument('--view-img', action='store_true', help='display results')
可以通过设置--view-img实时看到检测结果。一种方法是在命令行设置,另一种方法是在pycharm中直接对参数赋值。
pycharm菜单栏/运行/编辑配置。
parser.add_argument('--workers', type=int, default=0, help='maximum number of dataloader workers')
works参数强烈建议设置为0