用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。