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下的一些工作,后续会把精力放到结合硬件上面来做开发,有一些探索后,会继续博客分享。