Jetson agx xavier——pytorch下用于文本检测的字符区域识别
Xavier搭建pytorch等环境
参考我上一篇博客
环境要求
pytorch>=0.4.1
torchvision>=0.2.1
opencv-python>=3.4.2
前两个在之前的Xavier上都已经安装完成,opencv-python是刷机的时候会自己安装完成的,但是在anaconda里面访问不到(python import出错)。我后面重新刷了一遍程序,没有安装anaconda直接在安装pytorch核torchvision就可以使用。
下载git上的代码
当然还要下载数据集,也是在上面git的链接中,往下翻会找到如下图的界面,点击蓝色的click开始下载,任意下载一个就行。
下载完成后需要再下载一张带有文字的图片,新建文件夹放入(程序好像只能访问文件夹,无法直接访问图片)。
打开终端,按照自己下载的程序和数据集的路径,按照下面的格式输入。
python test.py --trained_model=[weightfile] --test_folder=[folder path to test images]
比如我的路径如下 ,第一个是程序里面test.py的路径,第二个数数据集的路径,第三个是存放图片的路径(注意输入的是文件夹名字不是文件名)
python /home/fzsc/Pytoch/CRAFT/test.py --trained_model=/home/fzsc/Pytoch/CRAFTmodel/craft_ic15_20k.pth --test_folder=/home/fzsc/Pytoch/CRAFT/figures
返回的结果在home下result文件夹中。
执行过程出现的错误
在执行的过程中可能会出现 “no module named skimage”,然后我在尝试直接“pip3 install scikit-image”之后也是出现一大片红色的失败提醒,忘了截图。。。。。
可以参考一下我的解决方法
先执行下面几个步骤
sudo pip3 install networkx
sudo pip3 install dask
sudo pip3 install scipy
最后再执行一下
pip3 install scikit-image
应该就没什么问题,执行的过程中有什么问题要一个个解决!