Skip to content

dongmingsun/tx2-erfpsp

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 

tx2-erfpsp

This repo is a tensorflow-version ERF-PSPNET implementation running on NVIDIA Jetson TX2, based on the papers below:

We use ZED Mini Stereo Camera to capture the road scene.

Prerequisites

  • Jetpack 3.1 (CUDA 8.0/cuDNN 6.0)
  • Python 3.5
  • ZED SDK 2.3

Requirements for Python

  • TensorFlow 1.4 (need to be compiled on TX2)
  • cv2
  • pyzed
  • opencv-python

Usage

python3 main.py

FYI, because the SpaceToBatchND op of TensorFlow has incorrect output on CUDA 9 (see https://devtalk.nvidia.com/default/topic/1044411/jetson-tx2/tensorflow-op-spacetobatchnd-does-not-work-correctly-on-tx2/post/5333429/), the code could only run with CUDA 8.0 (such that the only choice is Jetpack 3.1). Accordingly, the highest version that could be compiled smoothly on TX2 is tf 1.4.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages