はじめに Windowsで動くYoloを作っていたAlexeyABさんからYolov4が公開されました。また、ほぼ同じタイミングでUbuntu20.04がリリースされたので、この記事ではUbuntu20.04でYolov4を使ったオリジナルデータの学習を行います。自分の備忘録を兼ねて記事にしておきます。これから試される方の参考になれば嬉しいです。 今回は趣味の電子工作という観点から、下記のように電子部品を検出するモデルを作ります。 図は本記事の手順で実際に学習したモデルで推論した結果になります。 アルゴリズムの詳細についてはここでは触れませんので、ぜひ論文を読んで見ることをおすすめします。 環境構築 どこのご家庭にもある(?)下記のような構成で試しています。 ハードウェア Intel Corei5 8400 DDR4メモリ24GB Geforce RTX2070 Owltech SSR-6