utils.py 文件源码

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

项目:FastNeuralStyle 作者: bengxy 项目源码 文件源码
def excg_rgb_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

# Save model
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号