在jupyter里面循环运行py文件

import subprocess

# 参数定义
filenames = os.listdir(path)
params=[]
for i in filenames[1:]:
    ff = i.split('.')[0]
    ff = 'out/crop/'+ff+'/small_img1'
    params.append(ff)

# 循环运行

for p in params:
    cmd = ["python", "predict.py", "-path", p]
    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    out, err = proc.communicate()
    if proc.returncode != 0:
        print(f"Error running {cmd}: {err}")

其中,params为py文件可选参数。

猜你喜欢

转载自blog.csdn.net/weixin_49783489/article/details/130611414
今日推荐