せっかくなので自宅のGPUサーバー(Geforce750Tiなので残念GPUですが)で、VGG16をファインチューニングして文房具の判別をさせるべく学習させました。
今の所かなり的中率がよさそうです。
raspberry piで動かそうとしたところ、theanoではうまく動かず。
ビルド済みtensorflowではエラーで止まってしまうため、tensorflowをコンパイル中。
CPUが貧弱なので、時間がかかる。。。
コードと学習済みモデルとサンプル写真をgithubで共有します。
https://github.com/notfolder/stationery-predict.git
2019年5月10日金曜日
GW中の遊び(3)
せっかくなので、raspberry pi zeroにカメラをつけ、先に学習させた畳込みが3層のLeNetで白猫/キジ猫判定ができるデバイスを作ってみました。
被写体を写してボタンを押すと、推論して結果(キジならchai)を表示します。
被写体を写してボタンを押すと、推論して結果(キジならchai)を表示します。
GW中の遊び(2)
kerasで我が家の猫2匹の判別ができるようにDNNで学習させてみました。
http://aidiary.hatenablog.com/entry/20170110/1484057655
畳込みが3層のLeNetモデルで学習。
ほぼ100%判定できるようになったのですが、
白猫とキジ猫なので、画面真ん中に白いものがあれば白猫判定するという事に気付き。。。
http://aidiary.hatenablog.com/entry/20170110/1484057655
畳込みが3層のLeNetモデルで学習。
ほぼ100%判定できるようになったのですが、
白猫とキジ猫なので、画面真ん中に白いものがあれば白猫判定するという事に気付き。。。
2019年5月9日木曜日
登録:
投稿 (Atom)