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
![【Python】画像データ拡張ライブラリAlbumentationsを使ってみる - 旅行好きなソフトエンジニアの備忘録](https://cdn-ak-scissors.b.st-hatena.com/image/square/7592fba2add1505f238efe91c46d85fc4450f52a/height=288;version=1;width=512/https%3A%2F%2Fcdn-ak.f.st-hatena.com%2Fimages%2Ffotolife%2Fn%2Fni4muraano%2F20181120%2F20181120235803.jpg)