File size: 2,379 Bytes
5ea0d01 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | import os
def create_exp_folder():
# Step 1: 创建run文件夹（如果不存在）
if not os.path.exists("run"):
os.mkdir("run")
# Step 2: 创建train文件夹（如果不存在）
train_folder = os.path.join("run", "train")
if not os.path.exists(train_folder):
os.mkdir(train_folder)
# Step 3: 创建exp文件夹（检查是否存在）
exp_folder = os.path.join(train_folder, "exp")
if not os.path.exists(exp_folder):
os.mkdir(exp_folder)
os.mkdir(os.path.join(exp_folder, "weights")) # 创建weights文件夹
return exp_folder, os.path.join(exp_folder, "weights") # 返回exp和weights文件夹路径
# 如果exp文件夹已存在，则查找exp1, exp2, 等
exp_num = 1
while True:
# 动态命名exp1, exp2, ...
exp_folder_name = f"exp{exp_num}"
exp_folder = os.path.join(train_folder, exp_folder_name)
if not os.path.exists(exp_folder):
os.mkdir(exp_folder) # 创建新的exp文件夹
os.mkdir(os.path.join(exp_folder, "weights")) # 创建weights文件夹
return exp_folder, os.path.join(exp_folder, "weights") # 返回exp和weights文件夹路径
exp_num += 1 # 如果文件夹已存在，增加数字，继续查找下一个文件夹
def create_val_exp_folder():
# Step 1: 创建run文件夹（如果不存在）
if not os.path.exists("run"):
os.mkdir("run")
# Step 2: 创建train文件夹（如果不存在）
train_folder = os.path.join("run", "predict")
if not os.path.exists(train_folder):
os.mkdir(train_folder)
# Step 3: 创建exp文件夹（检查是否存在）
exp_folder = os.path.join(train_folder, "exp")
if not os.path.exists(exp_folder):
os.mkdir(exp_folder)
# 如果exp文件夹已存在，则查找exp1, exp2, 等
exp_num = 1
while True:
# 动态命名exp1, exp2, ...
exp_folder_name = f"exp{exp_num}"
exp_folder = os.path.join(train_folder, exp_folder_name)
if not os.path.exists(exp_folder):
os.mkdir(exp_folder) # 创建新的exp文件夹
return exp_folder # 返回新创建的文件夹路径
exp_num += 1 # 如果文件夹已存在，增加数字，继续查找下一个文件夹
|