在jetson tx2上使用pascal VOC0712训练caffe-ssd

折腾了两天，从绝望到光明~

新机器到能训练使用，大体过程如下：

1、刷机，参考我的最新博客： https://blog.csdn.net/maum61/article/details/97612542

2、安装配置caffe，完全参考这篇博客，没问题：https://blog.csdn.net/u012614287/article/details/81537743

3、安装配置caffe-ssd，按照这篇博客，没问题：https://github.com/weiliu89/caffe/tree/ssd

以上过程会经历一些缺少库的问题，自行百度解决吧，没有安装的就安装。

记录下来的一个是，可能会缺少openblas，使用 sudo apt-get install libopenblas-dev

4、我是从tensorflow转过来的，习惯了用pycharm，就在tx2上安装了pycharm，不是太顺利，主要还是对平台不熟悉，踩坑经历出查看我的博客：https://blog.csdn.net/maum61/article/details/98218802

5、使用pycharm打开caffe-ssd，注意，需要用root运行pycharm，不然会有很多意外惊喜。。

训练依然是参考https://github.com/weiliu89/caffe/tree/ssd ，但是在pycharm中运行某个.py文件，获得的路径就是该文件所在的目录，因此需要改一些路径，比如：

ssd_pascal.py里面的大概74行，是获取caffe_root的位置，也就是caffe_ssd的主目录，这里需要增加一句：

caffe_root = caffe_root+'/../../' 用来获取到caffe_root的位置。另外，tx2只有一个gpu，那么文件中大概330行左右，gpus="0,1,2,3"就得改成gpus="0"，不过经实际测试，这里把num_gpus=0，训练依然会在gpu 0上运行，不知道什么原因。可能是我在文件前面增加了这句话的缘故吧：CUDA_VISIBLE_DEVICES=0。

另外，在生成lmdb的时候，也要使用root权限，否则可能生成数据库不成功。默认是在~/data/下面的，下载的原始数据集和生成的lmdb都在这里，当程序在生成数据集的时候，~/data这个文件夹的大小是会变化的，最直观的是看free space一直在减小，这说明数据集lmdb生成成功。最后~/data这个文件夹的大小会在5.1G左右，而不是最初解压后的2.9G（我已经把下载到的压缩文件剪切到其他存储器上了，TX2存储空间捉襟见肘啊~）如果生成的数据库有问题，训练就会提示各种出错，奇葩的一个是（core dumped）。原因之一可能是TX2的内存不足，但是在改了batch_size和accum_batch_size都=2，还是吐核。那么问题就不是内存的事儿了，那就是数据库没有成功生成，查看一下lmdb文件，才十几kB，那显然是不对的。重新生成一下。

最后完事具备，开始自己训练VGGNET了，上个图，GPU利用率还是很高的，基本上都是在95%,风扇也是呼呼转。

在jetson tx2上使用pascal VOC0712训练caffe-ssd相关推荐

Jetson TX2上配置archiconda、Yolov5、tensorrtx环境问题记录
文章目录前言本文主要记录在Jetson TX2上配置archiconda.Yolov5.tensorrtx环境中遇到的问题以及解决方法.以及一些包的分享. 一.Jetson TX2刷机二.安装a ...
利用Python在Jetson TX2上抓取和显示摄像头影像
小编来自水下机器人社区193369905,里面小编给出了很多资料在本贴中,小编"我"分享了如何使用python 代码(及 OpenCV)在Jetson TX2上抓取和显示摄像头影 ...
在NVIDIA Jetson TX2上安装TensorFlow
五月两场 | NVIDIA DLI 深度学习入门课程 5月19日/5月26日一天密集式学习快速带你入门阅读全文> 正文共3070个字,3张图,预计阅读时间14分钟. 本文内容安排如下: 刷机 ...
FasterRcnn在Jetson TX2上测速
博主测试了在不同模式.精度下将FasterRCNN部署到Jetson TX2上的测速结果,与大家分享讨论. 对于MaskRCNN的部署结果可参见:MaskRCNN在Jetson tx2上的测速结果使 ...
Jetson tx2 上源码安装 pytorch1.0.0（真. 血泪史）
Jetson tx2 上源码安装 pytorch1.0.0(真. 血泪史) 本篇以在python3.5安装过程为例.在安装之前说明以下: 重点一:平台及cuda cudnn的安装问题 Jetson T ...
Jetson TX2 上安装Pycharm
Jetson TX2 上安装Pycharm Jetson TX2 上安装Pycharm Pycharm 2021.3.2 openjdk-11-jdk 安装pycharm中文语言包并将快捷方式锁定任务 ...
在 NVIDIA jetson tx2 上基于 realsense d435i 相机运行 vins 相关配置
在 NVIDIA jetson tx2 上基于 realsense d435i 相机运行 vins 相关配置一.tx2刷机二.安装librealsense,realsense驱动 1.安装依赖库 ...
jetson tx2上运行mobilenet-ssd的坑：interrupted by signal 9: SIGKILL
从ssd-caffe转战到mobilenet-ssd,也就是为了实时性.jetson tx2运行caffe-ssd前向的时间大概就是210ms.但是经过实际测试,对前5层卷积层使用CUDNN加速时,m ...
在jetson tx2上配置jetson-inference的流程以及报错解决
在显卡上运行完程序完成实验后,需要在tx2开发板上继续实现算法落地重拾前两天做的jetson-inference的学习,整理一下流程环境流程 #一些包的安装 $ sudo apt-get upda ...

在jetson tx2上使用pascal VOC0712训练caffe-ssd

在jetson tx2上使用pascal VOC0712训练caffe-ssd相关推荐

最新文章

热门文章