【目標檢測_3】基於tensorflow目標識別API執行已有的模型(運行demo coco數據集)

首先裝環境 參考鏈接

一 執行已經訓練好的模型 運行demo

1進入模型文件夾下的research文件夾中

進入research文件夾下!如果按照上面參考博客進行,那麼路徑可以由下面步驟查找:
首先確認tensorflow的路徑

python3
import tensorflow as tf
tf.__path__

查看tensorflow路徑
進入tensorflow文件夾,再依次進models_master/models(看自己下載的models名字命名)再進入research
在這裏插入圖片描述

2執行

jupyter-notebook

如果報錯則沒有安裝jupyter,重新開一個終端先裝jupyter

pip3 install jupyter

裝好之後重新進行:

jupyter-notebook

顯示如下:
在這裏插入圖片描述
點擊object_detection文件夾,進入之後再單擊下圖文件
在這裏插入圖片描述
效果如下:
在這裏插入圖片描述
如下圖操作:點擊單元格 運行所有
在這裏插入圖片描述
則會開始下載模型運行結果,運行速度與網速有關,因爲要下載模型,當下載完成之後可以看到object_detection文件下多了你下載的model
jupyter如何判斷運行到哪?
PS:單元格前面是星號表示還未運行或者還在運行,是數字沒有報錯則表示成功運行~ python的圓圈爲實心則在運行,否則運行結束
在這裏插入圖片描述

二 模型修改 _如何將例程中的ssd+mobilenet 改變爲faster 或者mask_rcnn

在這裏插入圖片描述
只需要將MODEL_NAME 修改就好模型類別列表鏈接
在這裏插入圖片描述
進入參考地址之後,往下翻,可以看到訓練模型表格如上所示,點擊就可以獲得全稱,就是加上更新日期的~直接複製修改再次運行即可 ,比如我這運行的是mask_rcnn 運行完成後結果爲:

在這裏插入圖片描述

三 如何添加自己的圖片進行識別

首先可以從object_detection_tutorial 中閱讀知道測試圖片路徑爲:PATH_TO_TEST_IMAGES_DIR = ‘test_images’
即本地的:~models-master/research/object_detection/test_image
在這裏插入圖片描述
圖片名字修改爲image3(image+序號)
TEST_IMAGE_PATHS = [ os.path.join(PATH_TO_TEST_IMAGES_DIR, ‘image{}.jpg’.format(i)) for i in range(1, 4) ]
其中的range(1,4)表示序號1,2,3因此添加圖片後此處範圍需要修改

在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章