當前位置：首頁 >

torchvision中Transform的normalize

發布時間：2023/12/20 53 豆豆

生活随笔收集整理的這篇文章主要介紹了 torchvision中Transform的normalize 小編覺得挺不錯的,現在分享給大家,幫大家做個參考.

RGB還是GBR?

根據[1][2],應該是:

RGB

transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])

第一個中括號是RGB的mean

第二個中括號是RGB的std

上述代碼的具體用法在[3]中有完整代碼.

Reference:

[1]https://discuss.pytorch.org/t/whats-the-channel-order-of-images-for-resnet-model-in-model-zoo/1017

[2]https://github.com/pytorch/vision/blob/master/torchvision/datasets/folder.py#L42

[3]https://github.com/appleyuchi/TPU_GPU_Code/blob/master/torchvision的用法實例.ipynb

總結

以上是生活随笔為你收集整理的torchvision中Transform的normalize的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。