【列表或字典在代码间的移植】

如何将列表存储至json文件再读取至其他代码中

在多模态任务中,常常需要使用预训练的模型去提取不同模态的特征信息,本代码可以搭建在代码内,直接使用源码来训练模型并获取表示,再将特征信息导入到多模态代码内进行后续的特征融合、推理等操作。

# 定义自己的list,将要存储的内容都存入list
list = []
list.append({'name': joey,
            'phon_enumber':66666})

# 将list存入json文件中
os.makedirs(json_file_dirpath, exist_ok=True)  # 创建文件夹
json_file_path = os.path.join(json_file_dirpath, 'entity_id2emb.json') # 设置文件名
with open(json_file_path, 'w+') as json_file:
    json.dump(list, json_file, indent=4, separators=(',', ': '))

# 将json文件读取回代码中
with open(json_file_path, 'r') as json_file:
    loaded_list = json.load(json_file)  # loaded_list即为存入的list

如果要存入list中的某部分元素为tensor张量,需要将其转化成为numpy数组再转化成列表list,再存入文件中,具体代码如下:

vector = vector.numpy().tolist()
vector = vector.cpu().detach().numpy().tolist()
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值