data_utils.py 文件源码

python
阅读 39 收藏 0 点赞 0 评论 0

项目:pytorch-fast-neural-style 作者: vishal1796 项目源码 文件源码
def batch_rgb_to_bgr(batch):
    batch = batch.transpose(0, 1)
    (r, g, b) = torch.chunk(batch, 3)
    batch = torch.cat((b, g, r))
    batch = batch.transpose(0, 1)
    return batch


# load image in RGB CxHxW [0,255]
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号