给图片添加一维数据

最新推荐文章于 2024-05-30 07:32:10 发布

jjw_zyfx

最新推荐文章于 2024-05-30 07:32:10 发布

阅读量509

点赞数

分类专栏： pytorch 文章标签： python

本文链接：https://blog.csdn.net/jjw_zyfx/article/details/121238128

版权

pytorch 专栏收录该内容

69 篇文章 8 订阅

订阅专栏

# Handle images with less than three channels
if len(img.shape) != 3: # 先判断了下是不是三维的
    # 不够三维添加一维，反正的弄够三维 img是一个Tensor对象
    img = img.unsqueeze(0)
    # 就是将最后一维的数据复制一分也用作新添加的那维的数据
    img = img.expand((3, img.shape[1:]))

优惠劵

jjw_zyfx

关注关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
给图片添加一维数据

# Handle images with less than three channelsif len(img.shape) != 3: # 先判断了下是不是三维的 # 不够三维添加一维，反正的弄够三维 img是一个Tensor对象 img = img.unsqueeze(0) # 就是将最后一维的数据复制一分也用作新添加的那维的数据 img = img.expand((3, img.shape[1:]))...
复制链接

扫一扫