はじめに
先日、【大人の盛り付けcom】というWEBサイトを作成した記事を投稿しましたが、
作成中に、機械学習の勉強をしてみたいと感じ、
【盛り付けレベルを判定してくれるAI】なるものを作成してみました。
次の見出しから、概要をご紹介します。
*しかし、ローカルでは動くのに、herokuにデプロイしたらエラーが出てしまう状況が続いております。
もし原因がお分かりの方いらっしゃりましたらアドバイスくださると嬉しいです🙏
サービス概要
普通クラス!!!
もう少し綺麗に盛り付けましょう!!!が出た
Twitterの方で動画あるので、是非みていただけたらと思います!
使用技術について
モデルの学習はGooglecolabを用いて行いました
画像投稿フォーム作成やviewsはこちらの中に記載のあったgithubを参考にしました!
限界
*今回、「最高クラス」「普通クラス」「もっときれいに盛り付けましょうクラス」と3つに分類するよう機械に学習させました。
20枚ずつ画像を使用しましたが、「最高クラス」が何度やっても出ないのです。。
普通クラスと最高クラスの見分けの精度がまだまだ不十分でした。
困ったこと
ローカルでは動くものの、デプロイ後にはエラーが出てしまう。
herokuにデプロイする際にPytorchの容量が大きすぎて全くデプロイが最初できなかった。
困って色々調べた結果、、自身がやってみて、容量を減らせたコツをブログにまとめたので、
同じくPytorchで今後herokuデプロイを考えている人は見てほしい
参考サイトはここ!
終わりに
初めて機械学習を使用したものを作ってみました。
学習させるところから、精度を上げていくところ、、、と本当に試行錯誤しながらでしたが、
機械学習がどのように行われているのか、を入門書で読みながら作ると良いかもしれないと感じました。
なかなかPytorchのネット記事はまだまだ少ないです。。
スクールの先生に教えてもらいながらでしたが、これからどんどん精度を上げていきたいと思いました。
そしてエラーの解消も・・・・