import subprocess


def main():
    data = 'D:\\syg\\yolov5\\datasets\\hqjvaj\\v1\\hqjvaj.yaml'
    project = 'D:\\syg\\yolov5\\runs\\hqjvaj\\train'
    name = 'v1'
    epochs = 10
    yolo_path = 'yolov5/train.py'

    subprocess.run(["python", yolo_path, "--data=" + data, "--project="
                        + project, "--name=" + name, "--epochs=" + str(epochs)], check=True)


if __name__ == "__main__":
    """
    如果此脚本被直接运行，则调用main()函数。
    如果此脚本被其他脚本导入，则不会自动调用main()。
    """
    main()