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]
data_utils.py 文件源码
python
阅读 39
收藏 0
点赞 0
评论 0
评论列表
文章目录