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

发布时间:2023年12月29日

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

  • 用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
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。