tf.split()函数的用法

本文介绍如何利用TensorFlow对图像进行分割操作,并展示了具体的代码实现过程。通过使用Python的PIL库加载图像,并将图像转换为NumPy数组,然后利用TensorFlow的split函数对图像的不同通道进行分割。

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

from PIL import  Image
import numpy as np
import tensorflow as tf

'''
    split 对维度进行分割
        tf.split( 
                data,  数据图片 ( 300*600*3)
                num_or_size_splits , 分割的数组 传个数  
                axis,  代表维度,当前的维度为  0  1 2
        )
        
        tf.split(data,3,2)   得到数据的维度为  [(300,600,1),(300,600,1),(300,600,1)]
        tf.split(data,[100.200,300],1)   得到数据的维度为  [(300,100,3),(300,100,3),(300,200,3)]
        
'''


img = Image.open('./test_data/tabby_cat.png')
img = np.array(img)

a = tf.split(img,3,2)


with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())

    a1  = sess.run([a ])

    # for i in range(len(a1[0])):
    #     print(a1[i].shape  )

    for i in range(len(a1[0])):
        # print(type(a1[0][i]))

        print(a1[0][i].shape)













# print(img.shape)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值