Skip to content
No description, website, or topics provided.
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
ckpt
README.md
color_definitions.py
main.py
model.py
resize.py
train.py
upsample.py

README.md

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.

You can’t perform that action at this time.