pytorch 如何使yolov7训练与1类从cocodataset

y53ybaqx 于 2022-12-04 发布在其他

关注(0)|答案(1)|浏览(222)

我正在处理yolov 7，train.py文件。我想使用cocodataset，但需要参加1个培训课程：我可以从www.example.com控制这个吗train.py?？

parser.add_argument('--single-cls', action='store_true', help='train multi-class data as single-class')

选项。但是我不知道如何使用此命令。
还有，列车日志上说;

tensorboard: Start with 'tensorboard --logdir runs/train', view at http://localhost:6006/

但佩奇什么也没说。

1条答案

如果你想使用所有80个类，你必须使用detect.py。通过train.py你可以训练你的自定义数据集来检测你的自定义对象。Tensorboard。首先，你必须安装

$ pip install tensorboard

如果您已经使用train.py进行了培训，则可以使用以下命令运行tensorboard：

$ tensorboard --logdir=runs/train

您可以在http：//localhost：6006/中查看结果