linux下faster rcnn入門一:配置網絡

linux下配置faster rcnn

流程參照博客readme
主要寫一些遇到問題

Zero

linux下anaconda下載tensorflow-gpu會配置對應的cuda以及cudnn,但是沒有cudahome以及一系列系統變量,所有需要再配置一下cuda以及cudnn,否則會報錯,找不到cudahome,前一篇博客寫了配置。

First

訓練模型下載失敗後,手動下載,放置位置
在這裏插入圖片描述

Second

https://blog.csdn.net/qq_34108714/article/details/89317004#commentsedit
識別出來的結果好多紅框重疊,後來在demo.py中調了閾值,發現毫無反應看了作者的issues才知道,我是一個gpu,gpu_id=0,看readme啊啊,以後遇到錯誤,要先看readme
GPU_ID=0
https://github.com/endernewton/tf-faster-rcnn/issues/235
Did you set the right GPU_ID? It should be 0 if you only have one GPU. I met the same error when I forget execute command “GPU_ID=0” before testing.

CUDA_VISIBLE_DEVICES=${GPU_ID} ./tools/demo.py
然後出現正確結果:
在這裏插入圖片描述備註:linux下一些修改可以參照windows下我前面寫的教程

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