numpy读取的时候进行img[:,:,::-1]这个操作就可以进行BGR转RGB的操作(读入是BGR, 否则就是RGB2BGR),这个不知道的话,穿越门写的非常清晰:
https://blog.csdn.net/u014453898/article/details/93716118
BGR转RGB并不是通道上第0通道和第二通道完全位置变换,而是最里层的3个值B和R进行转换,
再其次,torch里面的flip具有相同的功能,实现通道上的转换操作,
print("Forwarding", img.shape)# B,H,W,Cimg = torch.flip(img,[3])#BGR2RGB# img = img.float()mean=torch.tensor([123.675, 116.28, 103.53]).to(img)std=torch.tensor([58.395, 57.12, 57.375]).to(img)img = img - meanimg = img/std img = img.permute(0, 3, 1, 2).contiguous()x = self.img_backbone(img)