物体検出に使用するライブラリを mmdetection から transformers へ移行した話 - ANDPAD Tech Blog

序文 こんにちは、データ部ML Product Devチームに所属している酒巻です。 ML Product Devチームは「機械学習を活用した競合優位性のあるプロダクト開発」をミッションとし、プロダクト開発チームと協力して日々開発しています。 今回のブログでは、物体検出タスクで使用していた mmdetection から transformers へ移行した話について共有します。 背景 MLチームでは物体検出のタスクで mmdetection を様々なプロジェクトで活用しています。 しかし、直近1年以上更新のない状態が続いており、セキュリティ面において懸念が出てきました。 そこで mmdetection から別のライブラリへの移行を検討しました。 移行先候補 移行先として以下のようなものを候補として挙げました。 transformers Detectron2 super-gradients