2017-08-14 13 views
0

MacのUbuntu 16.04 ParallelsデスクトップでGoogleのtensorflowオブジェクト検出APIのjupyterノートブックを実行しています。オブジェクト検出タスクでバウンディングボックスの精度がどのように変化するかを確認するために、非デフォルトモデルの1つ(MobilenetでSSDでないモデル)をテストしたかったのです。Tensorflowオブジェクト検出APIのモデル動物園モデルの実行に関する問題

次のように私はノートにモデルの準備のセクションを変更:

# What model to download. 
MODEL_NAME = 'ssd_mobilenet_v1_coco_11_06_2017' 
MODEL_NAME = 'ssd_inception_v2_coco_11_06_2017' 
MODEL_NAME = 'rfcn_resnet101_coco_11_06_2017' 
#MODEL_NAME = 'faster_rcnn_resnet101_coco_11_06_2017' 
#MODEL_NAME = 'faster_rcnn_inception_resnet_v2_atrous_coco_11_06_2017' 


MODEL_FILE = MODEL_NAME + '.tar.gz' 
DOWNLOAD_BASE = 'http://download.tensorflow.org/models/object_detection/' 

# Path to frozen detection graph. This is the actual model that is used for the object detection. 
PATH_TO_CKPT = MODEL_NAME + '/frozen_inference_graph.pb' 

# List of the strings that is used to add correct label for each box. 
PATH_TO_LABELS = os.path.join('data', 'mscoco_label_map.pbtxt') 

NUM_CLASSES = 90 

を私は、メモリに凍結Tensorflowモデルをロードセルを実行するジャンプ。残念ながら、私は最後の3つのモデル(rfcn_resnet101_coco_11_06_2017、faster_rcnn_resnet101_coco_11_06_2017、faster_rcnn_inception_resnet_v2_atrous_coco_11_06_2017)、Firefoxでノートブックのクラッシュのいずれかを試してみて、私は、次のエラーメッセージが表示された場合:

The kernel appears to have died. It will restart automatically. 

は、だから私は最後の3つのモデルをテストすることができませんが私はtar.gzファイルをダウンロードしてobject_detectionフォルダに展開しましたが、誰かが私が間違っているかもしれないことを説明してもらえますか?

ありがとうございました!

+0

これはメモリの問題である可能性がありますか? –

答えて

1

この問題は、Parallelsに十分なメモリを割り当てられていなかったために発生したものです。このスクリプトは、より多くのメモリを割り当てた後に機能しました。先端のジョナサンに感謝します!

関連する問題