50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
from DataType.TypeDef import *
|
||
from DataType.ParamDef import *
|
||
|
||
|
||
def train(params: TrainParams):
|
||
"""
|
||
后台提交训练请求时，可以参考此方法，从前端传递的参数中取出需要的值
|
||
:param params:
|
||
:return:
|
||
"""
|
||
# 存在的参数可以使用name字段获取到
|
||
cpu_num = params.get("gpu_num")
|
||
print(cpu_num)
|
||
|
||
# 不存在的参数获取的时候返回None
|
||
th = params.get("th")
|
||
print(th)
|
||
|
||
|
||
def get_train_config_json():
|
||
"""
|
||
前端获取训练所需要的参数JSON字符串时，使用str方法即可获取JSON字符串
|
||
:return:
|
||
"""
|
||
# 第一种方法：
|
||
# 创建新对象
|
||
params = TrainParams()
|
||
# 从指定文件名中读取配置
|
||
params.read_from_file("TrainParams.json")
|
||
# 转为字符串
|
||
print(str(params))
|
||
|
||
# 第二种方法
|
||
# 也可以直接将文本传递给前端，但是可能需要考虑转义符号(\)的处理
|
||
with open("TrainParams.json", "r") as f:
|
||
print(f.read())
|
||
|
||
return params
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# 当前端需要获取参数列表时，参照此函数中的实现
|
||
get_train_config_json()
|
||
|
||
# 当前端传递回来参数的JSON字符串时，使用下述方法，将字符串反序列化为对象，然后传递给对应的函数
|
||
params = TrainParams()
|
||
params.read_from_str(
|
||
'["{\\"index\\": 0, \\"name\\": \\"gpu_num\\", \\"value\\": 2, \\"description\\": \\"\\", \\"default\\": 0, \\"type\\": \\"I\\"}", "{\\"index\\": 1, \\"name\\": \\"support_cpu\\", \\"value\\": true, \\"description\\": \\"\\", \\"default\\": false, \\"type\\": \\"B\\"}", "{\\"index\\": 2, \\"name\\": \\"labels\\", \\"value\\": 0, \\"description\\": \\"\\", \\"default\\": 0, \\"type\\": \\"E\\", \\"items\\": [\\"dog\\", \\"cat\\"]}"]')
|
||
train(params)
|