社会人研究者が色々頑張るブログ

pythonで画像処理やパターン認識をやっていきます

pytorchのお勉強(4):ネットワークの学習

はじめに

pytorchのお勉強の続きです。 今回は読み込んだデータセットを用いてモデルを学習していきます。
pytorchでCNNの学習システムを開発する際の全体像を以下に示します。

f:id:nsr_9:20210826195056p:plain

今回勉強する所はピンク色の矩形で囲んだ部分です。

続きを読む

pytorchのお勉強(3):オリジナルのデータセット読み込み

はじめに

pytorchのお勉強の続きです。 今回は学習するデータセットを読み取る所を学びます。

pytorchでCNNの学習システムを開発する際の全体像を以下に示します。
f:id:nsr_9:20210825161558p:plain

この画像で言うと、ピンクで囲った部分が前回までに勉強した部分で、青色の囲った部分が今回勉強する所です。

f:id:nsr_9:20210825161753p:plain

続きを読む

pytorchのお勉強(1):インストールとネットワークモデルの定義

はじめに

現代でComputer Visionの研究を行うならば、Deep Neural Network(DNN)の活用は必要不可欠になっています。
僕もその例に漏れず、普段からDNNのpythonライブラリを使用し研究活動を行っています。
DNNのpythonライブラリは様々なものがありますが、昨今では概ねモバイルやWeb用途にはtensorflow、学術研究用途にはpytorchが利用されている印象があります。
また、元々はライブラリ間での学習済みモデルの互換性がなかったのですが、ONNXというDNNの学習済みモデルの共通規格が登場し、相互変換が可能になりました。
OpenCVにもONNXモデルを利活用する機能が標準で搭載されているので、様々な応用が期待されますね。
ONNX | LearnOpenCV

僕は普段他のライブラリを使っているのですが、今回から1からpytorchを勉強していきたいと思います。
最終的には、今まで作った成果物と組み合わせて、より面白いことができたら良いなと思います。

続きを読む

Python+OpenCVによる前方車両検出(1)

はじめに

前回、"単眼車載カメラによる前方車両の検出”という、広島工業大学の紀要に掲載されていた論文を読みました。

nsr-9.hatenablog.jp

本手法は、走行空間モデルと横方向のエッジ画像の射影ヒストグラムを用いて、非学習ベース(CNNやSVMなどを使わない)で前方車両を検出する方法を提案していました。
今回からこの技術をpythonで実装していきます。

続きを読む