补全内容
This commit is contained in:
20
test.py
Normal file
20
test.py
Normal file
@ -0,0 +1,20 @@
|
||||
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()
|
Reference in New Issue
Block a user