min117の日記

初期desireもち。趣味Mac,メインFedora,仕事xp。

人工知能「Darknet」で画像内の物体識別を試す → 成功

画像の中に写ってる物体を識別する人工知能「Darknet」なるものがあるらしい。

f:id:min117:20200810113925p:plain

karaage.hatenadiary.jp

さっそくやってみる。

 

環境はiMac

f:id:min117:20200810114146p:plain

 

git clone https://github.com/pjreddie/darknet.git

f:id:min117:20200810105156p:plain

 

$ make

f:id:min117:20200810105234p:plain

 

30秒くらいで完了。

f:id:min117:20200810105300p:plain

 

$ cd darknet

$ ./darknet

f:id:min117:20200810105345p:plain

うむ。動く。

 

学習データ yolo.weights をダウンロードする。macOSには wget は入っていないので curl でやる。

$ curl -O http://pjreddie.com/media/files/yolo.weights

f:id:min117:20200810105506p:plain

 

$ ./darknet yolo 引数? cfg/設定ファイル 学習データ data/画像ファイル

$ ./darknet yolo test cfg/yolo.cfg yolo.weithts data/dog.jpg

f:id:min117:20200810111624p:plain

失敗した。cfg/yolo.cfgなんて設定ファイルがないというエラーになった。

 

cfgフォルダを覗いてみる。たしかに yolo.cfg なんてファイルは無い。

f:id:min117:20200810105727p:plain

 

よく分からんけど、yolo9000.cfgならあるので、これを試す。

$ ./darknet yolo test cfg/yolo9000.cfg yolo.weithts data/dog.jpg

f:id:min117:20200810105801p:plain

今度は実行はできた。predictions.png というファイルができるらしい

 

しかし、predictions.png なんてファイルは無い。

代わりに predictions.jpg があった。

 

開いてみる

$ open predictions.jpg

f:id:min117:20200810105917p:plain

ダメ。何も変わっていない。

 

このサイトを参考にさらに試す。

www.lisz-works.com

f:id:min117:20200810110232p:plain

まず、yolo.weightsがうまくダウンロードできてなかった。

 

curlは使わずに、リンクからSafariでダウンロードする。

http://pjreddie.com/media/files/yolo.weights

f:id:min117:20200810110316p:plain

200MB近くあれば正解。学習データなんだからこれくらいは重いはず。

 

次に、このサイトを参考にして実行コマンドを変える。

wakuphas.hatenablog.com

 

f:id:min117:20200810112204p:plain

 

$ ./darknet yolo test cfg/yolo.cfg yolo.weithts data/dog.jpg

 

$ ./darknet yolo detect cfg/yolov2.cfg yolo.weithts data/dog.jpg

$ ./darknet 引数? cfg/設定ファイル 学習データ data/画像ファイル

 

引数部分は「detect」が正しいらしい。

cfgは yolov2.cfg を使う。

 

$ ./darknet yolo detect cfg/yolov2.cfg yolo.weithts data/dog.jpg

あれ?動かない。

 

./darknet 直後のyoloを取っ払ったら動いた。

$ ./darknet detect cfg/yolov2.cfg yolo.weithts data/dog.jpg

f:id:min117:20200810112101p:plain

できた!

f:id:min117:20200810112634p:plain

 

こうなればあとはいろんな画像で試すだけ。例えばこんな。

f:id:min117:20200810112758p:plain

www.ghibli.jp

 

おお!

f:id:min117:20200810113433p:plain

Ohmu は流石に無理か。

 

こっちはどうか。

f:id:min117:20200810112832p:plain

 

f:id:min117:20200810113735p:plain

おお!

f:id:min117:20200810113826p:plain

素晴らしい。ただ解像度低いから?か姫様は認識できず。もう少し大きい画像で試してみる。

 

また観たくなった。映画館で見られるうちは何回でも観ておきたい。

 

参考

www.ghibri.net

 

wayohoo.com

f:id:min117:20200810120106p:plain