Ubuntu下FCN训练预测自己的数据集

1.labelme的安装,直接使用如下命令即可安装

pip install labelme

 2.由于用的是虚拟机,装的是tensorflow cpu版,版本可见我之前的博客语义分割之FCN训练预测自己的数据集_jiugeshao的专栏-CSDN博客

如果需要安装cuda和cuddn(非虚拟机环境下),可参见如下链接:

ubuntu18.04 安装CUDA11.0+cuDNN8.0.5+TensorFlow2.4.0_白羊007的博客-CSDN博客

Vmware 中 Ubuntu 20.04 安装 NVIDIA CUDA以及 cuDNN_Sizaif's 小屋-CSDN博客_虚拟机ubuntu安装cuda

 3.工程就是直接移植的我之前在windows下配置的FCN训练预测自己数据的代码

语义分割之FCN训练预测自己的数据集_jiugeshao的专栏-CSDN博客

4.由于之前在windows下,MITSceneParsing.pickle里含有\\字符,不识别,所以这边进行替换

修改BatchDatsetReader.py文件

5.运行FCN.py出现如下报错:

 

需要降低下numpy的版本

pip install numpy==1.16

 参数修改为train模式后,运行出现如下信息,表示配置成功,正在进行训练: 

 

到这篇博客,就先暂停Ubuntu下的一些工作,后续会把精力放到结合硬件上面来做开发,有一些探索后,会继续博客分享。

Guess you like

Origin blog.csdn.net/jiugeshao/article/details/120642735