图文识别(Faster R-CNN for Tensorflow)-草稿

环境情况:
1、ubuntu 18.04.3(虚拟环境)
2、python 3.6.8(如果有多个python,需要删除一些,并只保留一个)

https://www.tensorflow.org/install/source

1、安装TensorFlow

安装 Python 和 TensorFlow 软件包依赖项
   sudo apt install python-dev python-pip  # or python3-dev python3-pip
  安装 TensorFlow pip 软件包依赖项(如果使用虚拟环境,请省略 --user 参数):
   pip install -U --user pip six numpy wheel setuptools mock future>=0.17.1
   pip install -U --user keras_applications==1.0.6 --no-deps
   pip install -U --user keras_preprocessing==1.0.5 --no-deps

实际:

sudo apt install python3-dev python3-pip
pip3 install -U pip six numpy wheel setuptools mock future>=0.17.1
pip3 install -U --user keras_applications==1.0.6 --no-deps
pip3 install -U --user keras_preprocessing==1.0.5 --no-deps

2、安装Bazel

  • 步骤1:将Bazel发行版URI添加为包源
sudo apt install curl
curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -
echo "deb [arch=amd64] https://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
  • 步骤2:安装和更新Bazel
sudo apt update && sudo apt install bazel(无效)
sudo apt update && sudo apt install bazel-1.1.0(变更为此版本)
sudo apt update && sudo apt full-upgrade
sudo apt install openjdk-11-jdk

3、安装tensorflow

image.png

1、安装git

sudo apt install git
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow

2、配置编译系统

./configure

选是否的可以用 'n'

3、编译

bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package

变更为(上面那句受限于内存大小,会中途终止):

bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-msse4.1 --copt=-msse4.2 -k //tensorflow/tools/pip_package:build_pip_package --local_resources 3072,.5,1.0

当提示如下错误,但又明明安装了numpy时,可能是多个python版本造成的。

Traceback (most recent call last):
 File "<string>", line 1, in <module>
ModuleNotFoundError: No module named 'numpy'

编译时,可能会出现网络404的错误,多试几次,或尝试翻墙。

./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

pip install /tmp/tensorflow_pkg/tensorflow-2.0.0-cp36-cp36m-linux_x86_64.whl

PS: 1、最后的文件,看自己电脑中“tmp/tensorflow_pkg/”中的文件名
2、第二个命令执行时,需要pip进行在线下载,可能由于网络原因而中断,多试几次

此时会报错:

Could not find a version that satisfies the requirement grpcio>=1.8.6 (from tf-nightly==2.0.0) (from versions: none)

当出现“Successfully installed ....”则属于安装完成。

3 安装tf-faster-rcnn

需要先安装:

// pip安装cython
pip install --user --upgrade cython
// 安装cuda(不用安装,这个是gpu用的,虚拟机装不上)
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

下载并编译

git clone https://github.com/endernewton/tf-faster-rcnn.git
cd ./tf-faster-rcnn/lib
make clean
make

下载下来的版本,默认采用gpu进行编译(如果本身系统带gpu的,直接编译即可),需要做如下调整变更为cpu模式:

  • /lib/model/config.py
# Use GPU implementation of non-maximum suppression
__C.USE_GPU_NMS = False
  • lib/model/nms_wrapper.py
# --------------------------------------------------------
# Fast R-CNN
# Copyright (c) 2015 Microsoft
# Licensed under The MIT License [see LICENSE for details]
# Written by Ross Girshick
# --------------------------------------------------------
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from model.config import cfg
#from nms.gpu_nms import gpu_nms
from nms.cpu_nms import cpu_nms

def nms(dets, thresh, force_cpu=False):
  """Dispatch to either CPU or GPU NMS implementations."""

  if dets.shape[0] == 0:
    return []
  #if cfg.USE_GPU_NMS and not force_cpu:
    #return gpu_nms(dets, thresh, device_id=0)
  else:
    return cpu_nms(dets, thresh)
  • lib/setup.py


    image.png
image.png
image.png

4 安装 Python COCO API,这是为了使用COCO数据库

cd data
git clone https://github.com/pdollar/coco.git
cd coco/PythonAPI
make
cd ../../..

5 下载预训练模型

# Resnet101 for voc在07 + 12上进行了预训练设置 
./data/scripts/fetch_faster_rcnn_models.sh

faster_rcnn路径下执行

NET=res101
TRAIN_IMDB=voc_2007_trainval+voc_2012_trainval
mkdir -p output/${NET}/${TRAIN_IMDB}
cd output/${NET}/${TRAIN_IMDB}
ln -s ../../../data/voc_2007_trainval+voc_2012_trainval ./default
cd ../../..

运行示例程序(转到最下面)

GPU_ID=01
CUDA_VISIBLE_DEVICES=${GPU_ID} ./tools/demo.py

其他缺少的安装内容

pip install --user easydict
pip install --user opencv-python
pip install --user matplotlib

安装旧版本的tensorflow,新版本有个内容移除了,但r-cnn并没有变更,会报错

pip uninstall tensorflow
pip install --user tensorflow==1.13.2

6 结束

至此环境搭建完成,demo运行效果如下图:


image.png

参考地址:

1、https://docs.bazel.build/versions/master/install-ubuntu.html
2、https://www.tensorflow.org/install/source#setup_for_linux_and_macos

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 160,026评论 4 364
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,655评论 1 296
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,726评论 0 244
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,204评论 0 213
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,558评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,731评论 1 222
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,944评论 2 314
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,698评论 0 203
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,438评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,633评论 2 247
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,125评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,444评论 3 255
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,137评论 3 238
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,103评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,888评论 0 197
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,772评论 2 276
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,669评论 2 271

推荐阅读更多精彩内容