yolov5分割模型容易出错的点。

2023-12-29 10:32:19

离了个大谱，遇到这些问题真的是细节但浪费时间。列出来提示一下愚蠢的自己

  • 用lableme标注之后上传到roboflow生成数据集，并导出yolov5格式。当然也可以在roboflow上标注。

  • 出现dataset not find要注意yaml文件中的路径是否错误。比如val在路径中要写成valid。这是我觉得很扯淡的地方val:让我写成了val: ./dataset1/val/images，然后就一直报错找不到数据集。
    在这里插入图片描述

  • 还有要在import库之后加上os.environ["GIT_PYTHON_REFRESH"] = "quiet"

  • 要在main函数的第一行加上
    if __name__ == "__main__": os.environ["WANDB_API_KEY"] = '2565edddb46341b3aaed7a287d9318d4d756d633' opt = parse_opt()

  • 模型选择也要注意，条件反射选择yolov5s.pt，然而分割用到的是yolov5s-seg.pt。

文章来源:https://blog.csdn.net/chenyuan1111/article/details/135283555
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。