3.算法移植系列第三篇yolov5训练时修改train.py文件

染予

已于 2024-11-25 12:49:51 修改

阅读量176

点赞数 2

文章标签： YOLO

于 2024-11-11 12:03:10 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_52448289/article/details/143677906

版权

    parser = argparse.ArgumentParser()
    parser.add_argument("--weights", type=str, default="C:\\Users\\PC\PycharmProjects\\yolov5\\weights\\yolov5s.pt", help="initial weights path")
    parser.add_argument("--cfg", type=str, default="C:\\Users\\PC\\PycharmProjects\\yolov5\\models\\yolov5s.yaml", help="model.yaml path")
    parser.add_argument("--data", type=str, default="C:\\Users\\PC\\PycharmProjects\\yolov5\\dataset\\car\\data.yaml", help="dataset.yaml path")

同时修改\yolov5\\models\\yolov5s.yaml

修改为自己的nc种类 nc: 1 # number of classes