python 如何遍历图像

最新推荐文章于 2023-06-18 06:11:46 发布

Cv_Terry

最新推荐文章于 2023-06-18 06:11:46 发布

阅读量5.1k

点赞数

CC 4.0 BY-SA版权

分类专栏：学习python那些事深度不学习

本文链接：https://blog.csdn.net/ALZFterry/article/details/109512673

深度不学习同时被 2 个专栏收录

29 篇文章

订阅专栏

学习python那些事

16 篇文章

订阅专栏

本文介绍如何将1*2*256*256的Tensor数据转换为图像数据，并保存为txt格式。主要内容包括：使用numpy操作数据类型转换、调整数据维度顺序、以及遍历数据写入文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

pred 输入为 1* 2 * 256 * 256 的tensor

    pred = pred.numpy()#tensor转为数组
    pred = np.squeeze(pred)#把第一维去掉
    pred = np.transpose(pred, [1, 2, 0])#改变 维度顺序  256 * 256 * 2
    pred0=pred[:,:,0] 取第一通道
    pred0 = pred0 * 10
    pred0 = np.array(pred0, np.uint8)#opencv显示的话 要转换为8位无符号整型
    with open('pred0.txt', 'w') as q:#打开文件
        for i in range(pred0.shape[0]):#遍历行
            for w in range(pred0.shape[1]):#遍历列
            #for c in range（pred0.shape[2]）#这句话如果单通道不需要
                #if pred0[i,w]<8:
                #   print("notice",pred0[i,w])
                q.write("{0}   {1}  {2}\n".format(i,w,pred0[i,w]))