1. 程式人生 > >Tensorflow object detection環境搭建

Tensorflow object detection環境搭建

  • 為了復現這篇博文《Tensorflow object detection API 搭建屬於自己的物體識別模型(1)——環境搭建與測試》,需要安裝舊版本,如下:
  • Anaconda3 python3.5 (conda create -n tensorflow-gpu python=3.5) cuda8.0 cudnn6.0 tensorflow-gpu1.4 (pip install --ignore-installed --upgrade tensorflow-gpu==1.4)
  • 下載models不能是其最新版本,否則在執行model_builder_test.py時可能報錯,可以下載feature/kubeflow-testing版本
  • 安裝matplotlib2.1.1 (conda install matplotlib==2.1.1)
  • 安裝pillow4.3.0 (conda install pillow==4.3.0)
  • 在anaconda navigator中切換到環境tensorflow-gpu,然後開啟在Home頁面安裝jupyter notebook5.2.1(此處不用命令列安裝)