无人机下道路语义分割(FCN)

Posted on

环境

  • ubuntu16.4
  • CUDA 9.0
  • Anaconda 3 （numpy、os、datetime、matplotlib）
  • pytorch == 0.4.1 or 1.0
  • torchvision == 0.2.1
  • OpenCV-Python == 3.4.1

源码

https://github.com/qingzhouzhen/road_seg

原理

fcn

下采样之后不断上采样，每次上采样后和下采样后相同形状的张量相加，最终得到输入和输出张量的shape相同，每个像素都被预测类别。

数据

原图放在一个文件夹，标签图片放在另外一个文件夹，标签图片为黑白图片，但是是3通道。原图生成标签图片的脚本：https://github.com/qingzhouzhen/data_process/blob/master/generate_road_mask.py