20 lines
572 B
Python
20 lines
572 B
Python
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() |