训练YOLOv7-seg的问题

YOLOv7-SEG,也就是带分割的检测,运行train.py报图片中错误
之前一直能训练,今天突然打开就不行了

img

不知道你这个问题是否已经解决, 如果还没有解决的话:
  • 这篇博客: YOLOv5训练自己的数据集中的 3、修改train.py中的参数 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
     parser.add_argument('--epochs', type=int, default=200)  # 根据需要自行调节训练的epoch
     parser.add_argument('--batch-size', type=int, default=8)  # 根据自己的显卡调节,显卡不好的话,就调小点
     parser.add_argument('--cfg', type=str, default='models/yolov5s.yaml', help='*.cfg path') # 根据需要,自行选择模型
     parser.add_argument('--data', type=str, default='data/myvoc.yaml', help='*.data path')  # data设置为前两步中我们新建的myvoc.yaml
     parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='train,test sizes')  # 可调可不调
     parser.add_argument('--device', default='0', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')  # 使用CPU还是GPU训练

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^