kaggle の kernel で、 albumentations.torch が使えなくなった
結論
from albumentations.torch import ToTensor
ではなく、
from albumentations.pytorch import ToTensor
で呼び出さなくてはならない。
背景
画像関連の機械学習でよく使うデータ拡張をまとめた albumentations に、日頃からとてもお世話になっている。
検索すると色々出てくるし、公式のdocument も充実しているので、ここで使い方を紹介することはしない。ただ、pytorch と組み合わせて使う際に必要な処理の呼び出し方が変わっていたから、一応記事を書くことにした。
こんなのは、ちょっと調べれば出てくるし、1ヶ月後には常識になってるんだと思うけど、
Twitter に書くだけじゃ自分が忘れてしまいそうだったので。
参考
まさにtips だから、qiita にも後で上げておこう。