タグ

gpuとkerasに関するkoma_gのブックマーク (2)

  • 【物体検出】(GPU版)YOLO独自モデル簡単学習方法

    こんばんはエンジニアの眠れない夜です。 前回はkeras−yolo3の独自モデルの学習方法を紹介しました。 【物体検出】keras−yolo3の学習方法 手順が多くてハマりそうなポイントもちょいちょいあって、手間はかかりますが物体検出の独自モデルの学習ができるようになりましたね。 今回はGoogleColabを使ってGPU環境でYOLOの独自モデルを作ってみようと思います。 keras-yolo3で手作業で設定が必要な部分をどんどん置き換えていったらアノテーションファイルを作成してアップロードするだけで独自モデルの学習ができるようになりました\(^o^)/ もはやハマるところなし!! ノンプログラマーでも物体検出の独自モデルが作れそうな勢いです。Python初心者でも十分できます。 それでは一緒に手順を確認していきましょう。 【YOLO v3】独自モデルを学習させるまでの流れ 教師画像をダ

    【物体検出】(GPU版)YOLO独自モデル簡単学習方法
  • Python: Keras/TensorFlow で GPU のメモリを必要な分だけ確保する - CUBE SUGAR CONTAINER

    Keras のバックエンドに TensorFlow を使う場合、デフォルトでは一つのプロセスが GPU のメモリを全て使ってしまう。 今回は、その挙動を変更して使う分だけ確保させるように改めるやり方を書く。 環境には次のようにしてセットアップした Ubuntu 16.04 LTS を使っている。 blog.amedama.jp サンプルとして動作させるアプリケーションには Keras が提供している MNIST データセットを CNN で認識するものを使う。 まずはこれをダウンロードしておこう。 同時に、セッションをクリアするパッチも追加しておく。 $ wget https://raw.githubusercontent.com/fchollet/keras/master/examples/mnist_cnn.py $ echo 'K.clear_session()' >> mnist_c

    Python: Keras/TensorFlow で GPU のメモリを必要な分だけ確保する - CUBE SUGAR CONTAINER
  • 1