Ubuntu 16.04 tf-faster-rcnn 在CPU下训练数据

上一篇：Ubuntu 16.04 测试 tf-faster-rcnn 在CPU下运行

1、在根目录下打开终端，输入以下命令：

（1）如果是使用VGG16模型

mkdir -p data/imagenet_weights
cd data/imagenet_weights
wget -v http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz
tar -xzvf vgg_16_2016_08_28.tar.gz
mv vgg_16.ckpt vgg16.ckpt
cd ../..

（2）如果是Resnet101模型

mkdir -p data/imagenet_weights
cd data/imagenet_weights
wget -v http://download.tensorflow.org/models/resnet_v1_101_2016_08_28.tar.gz
tar -xzvf resnet_v1_101_2016_08_28.tar.gz
mv resnet_v1_101.ckpt res101.ckpt
cd ../..

2、下载voc2007数据集

http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

将三个文件下载解压到同一个文件夹下：data/VOCdevkit2007/

3、将tools/trainval_net.py拷贝到根目录下，并修改文件内容

（1）、注释：

  # if len(sys.argv) == 1:#   parser.print_help()#   sys.exit(1)

（2）添加default默认参数，省去在Edits Configurations输入命令参数

 parser.add_argument('--weight', dest='weight', default="./data/imagenet_weights/vgg16.ckpt",help='initialize with pretrained model weights',type=str)

输出模型路径可以修改以下参数的deault即可，代码默认生成的路径为：/output/default/voc_2007_trainval/default

parser.add_argument('--tag', dest='tag',help='tag of the model',default=None, type=str)

修改net参数，default="vgg16"

parser.add_argument('--net', dest='net',help='vgg16, res50, res101, res152, mobile',default='vgg16', type=str)

4、相关参数配置文件config.py

比如参数：__C.TRAIN.SNAPSHOT_ITERS就表示训练几次时保存一次

5、最后开始训练：

6、训练过程存在的问题

（1） keep_inds = np.append(fg_inds, bg_inds)

如果出现该情况解决办法就是删除缓存文件，比如默认缓存地址为：tf-faster-rcnn-master/data/cache

Ubuntu 16.04 tf-faster-rcnn 在CPU下训练数据相关推荐

Ubuntu 16.04 LTS　在anaconda环境下安装pytorch和tensorflow gpu，cuda10.0
文章目录准备更新笔记本的显卡驱动到较新的,最好是倒数第二新的版本安装cuda和cudnn: 安装conda: 安装tensorflow-gpu: 安装pytorch: 安装jupyter not ...
Android-x86-6.0定制之路 - 在Ubuntu 16.04.5上编译
前言由于下载的开源系统没有系统签名,所以不可能针对系统去做什么定制. 首先,要去下载 Android-x86-6.0 的系统源码,再去尝试编译系统,如果编译成功并且能够正常运行的话,才能研究去定制系 ...
Ubuntu 16.04下Caffe-SSD的应用（一）——编译Caffe-SSD的CPU版本与GPU版本
前言 1.之前用过Caffe的GooglNet做图像分类的相关项目,比起传统的办法,效果要好很多,在目标检测这块,之前也用传统的HOG加SVM实现过车辆,行人和马匹的检测,但正样本要上万张图像,才达到 ...
ubuntu 16.04 安装TensorFlow GPU版本
推荐新版安装教程 http://blog.csdn.net/chenhaifeng2016/article/details/78874883 在ubuntu 16.04上安装cuda8.0和cudnn ...
Ubuntu 16.04上安装SkyEye及测试
说明一下,在Ubuntu 16.04上安装SkyEye方法不是原创,是来自互联网,仅供学习参考. 一.检查支持软件包 gcc, make, vim(optional), ssh, subversion ...
Ubuntu 16.04 安装CUDA8.0+Cudnn6.0+TensorFlow+Caffe安装
参考博客: Ubuntu16.04 Caffe 安装步骤记录(超详尽):http://blog.csdn.net/yhaolpz/article/details/71375762 ubuntu16.0 ...
来自官方文档的Ubuntu 16.04 + tensorflow-GPU 配置
I Preprare for CUDA installation 官方文档:http://docs.nvidia.com/cuda/cuda-installation-guide-linux/ind ...
Ubuntu 16.04+1080Ti机器学习基本环境配置【转】
本文转载自:https://blog.csdn.net/MahoneSun/article/details/80808930 一.设置网络机器有两张网卡,将当前正在使用的"有线连接1&qu ...
ubuntu 16.04 1080Ti 安装Cuda
欢迎转载,请支持原创,保留原文链接:http://blog.ilibrary.me 原文已更新,用docker不需要这么繁琐的步骤.2019.08.13 前段时间做自动摘要和自动翻译,跑深度学习的算法 ...

Ubuntu 16.04 tf-faster-rcnn 在CPU下训练数据

Ubuntu 16.04 tf-faster-rcnn 在CPU下训练数据相关推荐

最新文章

热门文章