xkrs_manan/RODY
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

打印data_list

Browse Source
This commit is contained in:
552068321@qq.com 2022-11-07 17:02:19 +08:00
parent df0581b77a
commit a4227b57fb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/controller/AlgorithmController.py
View File

@ -249,7 +249,7 @@ def train_R0DY(params_str, id):
params.read_from_str(params_str) params.read_from_str(params_str)
print(params.get('device').default) print(params.get('device').default)
data_list = file_tool.get_file(ori_path=params.get('DatasetDir').value, type_list=params.get('classname').value) data_list = file_tool.get_file(ori_path=params.get('DatasetDir').value, type_list=params.get('classname').value)
print(data_list) print('---------------------',data_list)
weights = params.get('resumeModPath').value # 初始化模型绝对路径 weights = params.get('resumeModPath').value # 初始化模型绝对路径
img_size = params.get('img_size').value img_size = params.get('img_size').value
savemodel = os.path.splitext(params.get('saveModDir').value)[0] + '_' + str(img_size) + '.pt' # 模型命名加上图像参数 savemodel = os.path.splitext(params.get('saveModDir').value)[0] + '_' + str(img_size) + '.pt' # 模型命名加上图像参数