Faster R-CNNをpytorchでサクッと動かしてみた - Qiita

記事の更新 動かしやすいようにcolab実装版を書きました。 ぜひ以下の記事をご参照ください。 （dataloaderが速くなった） 2022/4/26更新 学習,推論,評価基盤を整えて，バグの修正を行ったコードを公開しました． 実験を行う際はこちらのコードを使用することをお勧めします． https://github.com/Ryunosuke-Ikeda/Faster-R-CNN-pytorch はじめに Faster R-CNNをちゃんとしたデータセットで動かしている記事が少なくてかなり苦労したから備忘録 初めての記事投稿なので至らないところもあるとは思いますが何か間違い等ありましたらご指摘をお願いします。 諸注意 ※本記事はPSCAL VOCフォーマットのデータセット向けです。 私はBDD100KというデータセットをPascalVOCフォーマットに変換して学習を行ったためclassラ