打印data_list

This commit is contained in:
552068321@qq.com 2022-11-07 16:59:19 +08:00
parent 50a4898801
commit df0581b77a

View File

@ -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' # 模型命名加上图像参数