2021-11-23から1日間の記事一覧

【PyTorch】detectron2を使って独自データに対して物体検出モデルの学習を行う

はじめに 学習データ Pythonスクリプト 結果 推論 さいごに はじめに前回推論を行いました。 touch-sp.hatenablog.com 今回は独自データに対して物体検出モデルの学習を行ってみたいと思います。学習データこちらで作成したデータ(VOCフォーマット)を使用…

深層学習でGPUのメモリ不足に直面したら

はじめにDeep Learningの学習にはGPUのメモリが必要になります。 不足に直面した時に実際試して効果がありそうと感じた方法をいくつか紹介します。 他のソフトウェア(ゲームやブラウザなど)に影響がありますのであくまで用途を限定して参考にしてください…