PyTorch版のYOLO v3を作っている人がいたので試してみようと思っています。 github.com ただ、Trainにデータ拡張が入っていないのでデータ拡張ロジックを追加したいと思ったところ、 Albumentationsというライブラリを見つけました。 github.com 物体検出やセグメンテーションにも利用可能そうなので早速試してみました。 使い方は以下を実施すれば良さげです。 Composeを作って、Composeの中に実施したいデータ拡張を記述 Composeに画像、ラベル、クラスIDを含むディクショナリを投入 以下はComposeを作るコードになります。 from albumentations import Compose from albumentations.augmentations.transforms import Resize, HorizontalFlip