せっかくなので自宅のGPUサーバー(Geforce750Tiなので残念GPUですが)で、VGG16をファインチューニングして文房具の判別をさせるべく学習させました。
今の所かなり的中率がよさそうです。
raspberry piで動かそうとしたところ、theanoではうまく動かず。
ビルド済みtensorflowではエラーで止まってしまうため、tensorflowをコンパイル中。
CPUが貧弱なので、時間がかかる。。。
コードと学習済みモデルとサンプル写真をgithubで共有します。
https://github.com/notfolder/stationery-predict.git
0 件のコメント:
コメントを投稿