2019年5月10日金曜日

GW中の遊び(4)

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

0 件のコメント:

コメントを投稿