打印data_list
This commit is contained in:
parent
50a4898801
commit
df0581b77a
@ -249,6 +249,7 @@ def train_R0DY(params_str, id):
|
||||
params.read_from_str(params_str)
|
||||
print(params.get('device').default)
|
||||
data_list = file_tool.get_file(ori_path=params.get('DatasetDir').value, type_list=params.get('classname').value)
|
||||
print(data_list)
|
||||
weights = params.get('resumeModPath').value # 初始化模型绝对路径
|
||||
img_size = params.get('img_size').value
|
||||
savemodel = os.path.splitext(params.get('saveModDir').value)[0] + '_' + str(img_size) + '.pt' # 模型命名加上图像参数
|
||||
|
Loading…
Reference in New Issue
Block a user