YOLOv7与YOLOv5 训练自己的数据集

WVV_SOYEON

已于 2024-04-22 15:01:11 修改

阅读量387

点赞数 3

文章标签： YOLO 人工智能深度学习

于 2024-04-18 14:46:15 首次发布

本文链接：https://blog.csdn.net/wvv_soyeon/article/details/137920295

版权

本文分享了作者在服务器端使用YOLOv7进行训练时遇到的问题，包括权重文件名的修正、文件路径的设置，以及如何配置数据集和处理可能的内存问题。建议对YOLOv7初学者有参考价值。

摘要由CSDN通过智能技术生成

对自己YOLOv7以及YOLOv5训练自己数据集的一些碎碎念，毫无技术含量，大佬劝退！大佬劝退！大佬劝退！

目前是在服务器端训练v7，想起什么写什么吧……

1.YOLOv7源代码运行可有意思，权重文件的名称下载下来都是有问题的……

如果要运行v7的train.py，记得改代码中的权重文件名称如下：

（直接下载的权重文件名称是yolo7 .pt，但是权重文件是yolov7.pt，记得改！并且yolov7.pt和yolov7 .pt是不一样的！）

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--weights', type=str, default='yolo7.pt', help='initial weights path')

2.文件引用也有意思，相对路径我自己本机跑的话是会出错的，是找不到文件的。改成绝对路径就不报错，找到文件，可以正常运行了，好欸！（如果使用的是pycharm的话记得路径是双斜杠例如：E:\\20240412）

在服务器跑的话就只能用相对路径了，为了避免它找不到相对应的文件，建议直接把需要的文件拖在根目录下。这样就能找到了。请看下图：

train.py中的路径也要根据文件存放位置修改，文件放置请看上图：

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--weights', type=str, default='yolo7.pt', help='initial weights path')
    parser.add_argument('--cfg', type=str, default='yolov7.yaml', help='model.yaml path')
    parser.add_argument('--data', type=str, default='data_yolov7.yaml', help='data.yaml path'