将科学计数法表示的数组(张量)转化为十进制表示,以输出.pth特征文件里面的空间特征为例

我想要查看.pth特征文件里面的内容,结果输出的数据都是科学计数法表示的,如下
在这里插入图片描述

具体数字多大看不出来,加上下面一句:

np.set_printoptions(formatter={'float_kind':'{:f}'.format})

输出结果如下:
在这里插入图片描述

代码

import numpy as np
import torch
import sys
import os


class Logger(object):
    def __init__(self, filename="Default.log"):
        self.terminal = sys.stdout
        self.log = open(filename, "a")

    def write(self, message):
        self.terminal.write(message)
        self.log.write(message)

    def flush(self):
        pass

np.set_printoptions(threshold=np.inf)

sys.stdout = Logger('c.txt')

state_dict = torch.load("data/rcnn_feature/9.pth")  # 这个是特征文件,注意路径
s = state_dict["spatial_feature"] # s是numpy类型的
# print(state_dict["image_feature"].shape)
np.set_printoptions(formatter={'float_kind':'{:f}'.format})

print(s)

print('---------------------------------------')

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值