KittiSeg Road Segmentation model implemented in tensorflow.

GITHUB:https://github.com/MarvinTeichmann/KittiSeg

git clone https://github.com/MarvinTeichmann/KittiSeg.gi
cd KittiSeg
git submodule update --init --recursive

python download_data.py --kitti_url  http://kitti.is.tue.mpg.de/kitti/data_road.zip

python3 demo.py --input_image data/demo/demo.png

自己找了几张图测试,效果还行。

 

待解决:

  1. 对非公路如石子路识别不理想。
  2. 公路车多的时候,检测不出公路
  3. 识别的时候,似乎没有使用GPU。
    1. 原因是我机器上装了 tensorflow cpu 和gpu 版,gpu 是 1.2  cpu 是1.4 这个项目在1.2上跑不起来。都卸了重装最新版的 tensorflow-gpu。
    2. 安装的时候报错。
    3.  libcudnn.so.6: cannot open shared object file: No such file or directory
    4. https://developer.nvidia.com/rdp/cudnn-download 下载新的 cudnn 安装。
    5. Navigate to your <cudnnpath> directory containing the cuDNN Tar file.
    6. Unzip the cuDNN package.
      $ tar -xzvf cudnn-9.0-linux-x64-v7.tgz
    7. Copy the following files into the CUDA Toolkit directory.
      $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
      $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
      $ sudo chmod a+r /usr/local/cuda/include/cudnn.h
      /usr/local/cuda/lib64/libcudnn*
      
      
    8.  sudo ldconfig /usr/local/cuda/lib64  #没有这步,还会出错。
    9. 2017-12-15 12:25:14.185527: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1030] Found device 0 with properties:
      
      name: GeForce GTX 1080 Ti major: 6 minor: 1 memoryClockRate(GHz): 1.62
      
      pciBusID: 0000:01:00.0
      
      totalMemory: 10.90GiB freeMemory: 10.46GiB
      
      2017-12-15 12:25:14.185550: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1120] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX 1080 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1)
    10. 问题解决了

《KittiSeg Road Segmentation model implemented in tensorflow.》上有3条评论

  1. 您好!我在测试这个程序的时候遇到了问题,想请教您一下。
    遇到的问题:
    File “demo.py”, line 51, in
    from seg_utils import seg_utils as seg
    ModuleNotFoundError: No module named ‘seg_utils’
    找不到seg_utils在源文件中也没有,网上也没找到。

发表评论

电子邮件地址不会被公开。

This site uses Akismet to reduce spam. Learn how your comment data is processed.