Tensorflow MobileNet移植到Android

1 CKPT模型轉換pb文件 使用上一篇博客《MobileNet V1官方預訓練模型的使用》中下載的MobileNet V1官方預訓練的模型《MobileNet_v1_1.0_192》。雖然打包下載的文件中包含已經轉換過的pb文件,但是官方提供的pb模型輸出是1001類別對應的概率,我們需要的是概率最大的3類。可在原始網絡中使用函數tf.nn.top_k獲取概率最大的3類,將函數tf.nn.top
相關文章
相關標籤/搜索